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

switch from case to ifelse

Browse Source
This commit is contained in:
Max Bain
2025-01-11 17:11:21 +00:00
parent c288f4812a
commit aaddb83aa5
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
whisperx/asr.py
View File

@ -397,11 +397,12 @@ def load_model(
print("Use manually assigned vad_model. vad_method is ignored.")
vad_model = vad_model
else:
match vad_method:
case "silero":
vad_model = whisperx.vads.Silero(**default_vad_options)
case "pyannote" | _:
vad_model = whisperx.vads.Pyannote(torch.device(device), use_auth_token=None, **default_vad_options)
if vad_method == "silero":
vad_model = whisperx.vads.Silero(**default_vad_options)
elif vad_method == "pyannote":
vad_model = whisperx.vads.Pyannote(torch.device(device), use_auth_token=None, **default_vad_options)
else:
raise ValueError(f"Invalid vad_method: {vad_method}")
return FasterWhisperPipeline(
model=model,