From 6bb2f1cd488236776d3a3d5c7df8b058576b9b46 Mon Sep 17 00:00:00 2001 From: Full Name Date: Mon, 1 Jan 2024 14:56:51 +0530 Subject: [PATCH] Added Vad custom option --- whisperx/asr.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/whisperx/asr.py b/whisperx/asr.py index dba8271..ac816ff 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -262,6 +262,7 @@ def load_model(whisper_arch, compute_type="float16", asr_options=None, language : Optional[str] = None, + vad_model=None, vad_options=None, model : Optional[WhisperModel] = None, task="transcribe", @@ -337,7 +338,10 @@ def load_model(whisper_arch, if vad_options is not None: default_vad_options.update(vad_options) - vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options) + if vad_model is not None: + vad_model = vad_model + else: + vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options) return FasterWhisperPipeline( model=model,