add base option

k2-fsa · Apr 30, 2024 · 844ca42 · 844ca42
1 parent c292dc6
commit 844ca42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/multi_zh-hans/ASR/whisper/decode.py b/egs/multi_zh-hans/ASR/whisper/decode.py
@@ -215,7 +215,7 @@ def get_parser():
         "--model-name",
         type=str,
         default="large-v2",
-        choices=["large-v2", "large-v3", "medium", "small", "tiny"],
+        choices=["large-v2", "large-v3", "medium", "base", "small", "tiny"],
         help="""The model name to use.
         """,
     )