Merge pull request #33 from linto-ai/bugfix/whisper-large-v3

Support of new Whisper model large-v3

Author: Jeronymous

.envdefault

@@ -13,7 +13,7 @@ BROKER_PASS=
 # STT MODELING PARAMETERS
 ############################################
 
-# The model can be a path to a model, or a model name ("tiny", "base", "small", "medium", "large-v1" or "large-v2")
+# The model can be a path to a model, or a model name ("tiny", "base", "small", "medium", "large-v1", "large-v2" or "large-v3")
 MODEL=medium
 
 # The language can be in different formats: "en", "en-US", "English", ...

RELEASE.md

@@ -1,3 +1,6 @@
+# 4.0.5
+- Support of Whisper large-v3 model
+
 # 4.0.4
 - Add integration of Whisper models from transformers
 - Add support of prompt from Whisper models (env variable PROMPT)

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

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