Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 14:35:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
mercury233 2024-10-10 11:38:06 +08:00
parent a9c001394a
commit 0390c216d6
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
whisper/utils.py
View File

@ -302,22 +302,20 @@ def get_writer(
"json": WriteJSON, "json": WriteJSON,
} }
if output_format == "all":
all_writers = [writer(output_dir) for writer in writers.values()]
def write_all(
result: dict, file: TextIO, options: Optional[dict] = None, **kwargs
):
for writer in all_writers:
writer(result, file, options, **kwargs)
return write_all
writer_funcs = [] writer_funcs = []
for format in output_format.split(','): if output_format == "all":
if format in writers: writer_funcs = [writer(output_dir) for writer in writers.values()]
writer_funcs.append(writers[format]) else:
else: for format in output_format.split(','):
raise ValueError(f"Output format '{format}' is not supported.") if format in writers:
writer_funcs.append(writers[format](output_dir))
else:
raise ValueError(f"Output format '{format}' is not supported.")
return lambda result, file, options: [writer(output_dir)(result, file, options) for writer in writer_funcs] def write_all (
result: dict, file: TextIO, options: Optional[dict] = None, **kwargs
):
for writer in writer_funcs:
writer(result, file, options, **kwargs)
return write_all