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

support comma separated output_format

Browse Source
This commit is contained in:
mercury233 2023-03-07 09:59:13 +08:00
parent 25639fc17d
commit 7f0b03461a
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
whisper/utils.py
View File

@ -313,4 +313,11 @@ def get_writer(
return write_all
return writers[output_format](output_dir)
writer_funcs = []
for format in output_format.split(','):
if format in writers:
writer_funcs.append(writers[format])
else:
raise ValueError(f"Output format '{format}' is not supported.")
return lambda result, file: [writer(output_dir)(result, file) for writer in writer_funcs]