feat: update versions for pyannote:3.3.2 and faster-whisper:1.1.0 (#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 <31799121+Barabazs@users.noreply.github.com>
This commit is contained in:
Icaro Bombonato
2024-12-31 06:41:09 -03:00
committed by GitHub
parent 7fdbd21fe3
commit 15ad5bf7df
3 changed files with 3 additions and 2 deletions

View File

@ -1,6 +1,6 @@
torch>=2
torchaudio>=2
faster-whisper==1.0.3
faster-whisper==1.1.0
ctranslate2==4.4.0
transformers
pandas

View File

@ -21,7 +21,7 @@ setup(
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"],
},

View File

@ -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,