feat: update versions for pyannote:3.3.2 and faster-whisper:1.1.0 (m-…

…bain#936) * chore: bump faster-whisper to 1.1.0 * chore: bump pyannote to 3.3.2 * feat: add multilingual option in load_model function --------- Co-authored-by: Barabazs <[email protected]>
opus-pro · Jan 2, 2025 · 3324984 · 3324984
1 parent 44299f1
commit 3324984
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 torch>=2
 torchaudio>=2
-faster-whisper==1.0.3
+faster-whisper==1.1.0
 ctranslate2==4.4.0
 transformers
 pandas

diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
             open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
         )
     ]
-    + [f"pyannote.audio==3.1.1"],
+    + [f"pyannote.audio==3.3.2"],
     entry_points={
         "console_scripts": ["whisperx=whisperx.transcribe:cli"],
     },

diff --git a/whisperx/asr.py b/whisperx/asr.py
@@ -319,6 +319,7 @@ def load_model(whisper_arch,
         "word_timestamps": False,
         "prepend_punctuations": "\"'“¿([{-",
         "append_punctuations": "\"'.。,，!！?？:：”)]}、",
+        "multilingual": model.model.is_multilingual,
         "suppress_numerals": False,
         "max_new_tokens": None,
         "clip_timestamps": None,