From 7f0b03461a77e5b3012be078b3a4275b64f68dc1 Mon Sep 17 00:00:00 2001 From: mercury233 Date: Tue, 7 Mar 2023 09:59:13 +0800 Subject: [PATCH] support comma separated output_format --- whisper/utils.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/whisper/utils.py b/whisper/utils.py index 9b9b138..e793027 100644 --- a/whisper/utils.py +++ b/whisper/utils.py @@ -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]