Support of new Whisper model large-v3

linto-ai · Nov 13, 2023 · 97320ce · 97320ce
1 parent 5a2bb1b
commit 97320ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/requirements.ctranslate2.txt b/requirements.ctranslate2.txt
@@ -10,6 +10,6 @@ pyyaml>=5.4.1
 requests>=2.26.0
 wavio>=0.0.4
 websockets
-ctranslate2==3.18.0
-#faster_whisper==0.7.1
-git+https://github.com/linto-ai/faster-whisper.git@d9cffcaad763def754124977cc66150f0efcd7ea
+#faster_whisper==0.10.0
+# This is version faster_whisper==0.9.0 + prompt propagation + fix for large-v3
+git+https://github.com/linto-ai/faster-whisper.git@aad9e7508b528e79be2a9975ac79ef8317f02a6d
diff --git a/stt/processing/load_model.py b/stt/processing/load_model.py
@@ -52,7 +52,7 @@ def load_whisper_model(model_type_or_file, device="cpu", download_root=None):
             device = "cuda"
 
         if not os.path.isfile(os.path.join(model_type_or_file, "model.bin")) and \
-            model_type_or_file not in ["tiny.en", "tiny", "base.en", "base", "small.en", "small", "medium.en", "medium", "large-v1", "large-v2"]:
+            not max([model_type_or_file.startswith(prefix) for prefix in ["tiny", "base", "small", "medium", "large"]]):
 
             # Convert transformer model