Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +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

24
whisper/utils.py
View File

@ -302,22 +302,20 @@ def get_writer(
"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 = []
if output_format == "all":
writer_funcs = [writer(output_dir) for writer in writers.values()]
else:
for format in output_format.split(','):
if format in writers:
writer_funcs.append(writers[format])
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