From 6be02cccfa88ff7a9dd0bcea335e0e5bf1fcda03 Mon Sep 17 00:00:00 2001 From: Max Bain <36994049+m-bain@users.noreply.github.com> Date: Mon, 13 Jan 2025 10:08:09 +0000 Subject: [PATCH] Update asr.py --- whisperx/asr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/whisperx/asr.py b/whisperx/asr.py index c75144b..b0aa7b4 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -305,7 +305,7 @@ def load_model( asr_options: Optional[dict] = None, language: Optional[str] = None, vad_model = None, - vad_method = None, + vad_method: str = "pyannote", vad_options: Optional[dict] = None, model: Optional[WhisperModel] = None, task="transcribe", @@ -399,7 +399,7 @@ def load_model( else: if vad_method == "silero": 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) else: raise ValueError(f"Invalid vad_method: {vad_method}")