chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update asr.py

Browse Source
This commit is contained in:
Max Bain
2025-01-13 10:08:09 +00:00
committed by GitHub
parent 1ec527375a
commit 6be02cccfa
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisperx/asr.py
View File

@ -305,7 +305,7 @@ def load_model(
asr_options: Optional[dict] = None, asr_options: Optional[dict] = None,
language: Optional[str] = None, language: Optional[str] = None,
vad_model = None, vad_model = None,
vad_method = None, vad_method: str = "pyannote",
vad_options: Optional[dict] = None, vad_options: Optional[dict] = None,
model: Optional[WhisperModel] = None, model: Optional[WhisperModel] = None,
task="transcribe", task="transcribe",
@ -399,7 +399,7 @@ def load_model(
else: else:
if vad_method == "silero": if vad_method == "silero":
vad_model = whisperx.vads.Silero(**default_vad_options) vad_model = whisperx.vads.Silero(**default_vad_options)
elif vad_method == "pyannote" or not vad_method: elif vad_method == "pyannote":
vad_model = whisperx.vads.Pyannote(torch.device(device), use_auth_token=None, **default_vad_options) vad_model = whisperx.vads.Pyannote(torch.device(device), use_auth_token=None, **default_vad_options)
else: else:
raise ValueError(f"Invalid vad_method: {vad_method}") raise ValueError(f"Invalid vad_method: {vad_method}")