Added changes

m-bain · Jan 4, 2024 · 2760815 · 2760815
1 parent 4bc12cb
commit 2760815
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/whisperx/asr.py b/whisperx/asr.py
@@ -262,7 +262,7 @@ def load_model(whisper_arch,
                compute_type="float16",
                asr_options=None,
                language : Optional[str] = None,
-               vad_model=None,
+               vad_model_fp=None,
                vad_options=None,
                model : Optional[WhisperModel] = None,
                task="transcribe",
@@ -339,8 +339,8 @@ def load_model(whisper_arch,
     if vad_options is not None:
         default_vad_options.update(vad_options)
 
-    if vad_model is not None:
-        vad_model = vad_model
+    if vad_model_fp is not None:
+        vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options, model_fp=vad_model_fp)
     else:
         vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options)