vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix vad_method is none

Browse Source
This commit is contained in:
winking324
2025-01-13 13:53:35 +08:00
parent 6695426a85
commit 1ec527375a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisperx/asr.py
+1 -1
View File
@@ -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": elif vad_method == "pyannote" or not vad_method:
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}")