Add support for large-v3-turbo model

Softcatala · Oct 10, 2024 · a4f061e · a4f061e
1 parent 71256bd
commit a4f061e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/whisper_ctranslate2/commandline.py b/src/whisper_ctranslate2/commandline.py
@@ -14,6 +14,7 @@
     "large-v1",
     "large-v2",
     "large-v3",
+    "large-v3-turbo",
     "distil-large-v2",
     "distil-large-v3",
     "distil-medium.en",

diff --git a/src/whisper_ctranslate2/whisper_ctranslate2.py b/src/whisper_ctranslate2/whisper_ctranslate2.py
@@ -112,6 +112,9 @@ def main():
     hf_token = args.pop("hf_token")
     speaker_name = args.pop("speaker_name")
 
+    if model == "large-v3-turbo":
+        model = "deepdml/faster-whisper-large-v3-turbo-ct2"
+
     language = get_language(language, model_directory, model)
     options = get_transcription_options(args)