Skip to content

Commit 2760815

Browse files
Full NameFull Name
Full Name
authored and
Full Name
committed
Added changes
1 parent 4bc12cb commit 2760815

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

whisperx/asr.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def load_model(whisper_arch,
262262
compute_type="float16",
263263
asr_options=None,
264264
language : Optional[str] = None,
265-
vad_model=None,
265+
vad_model_fp=None,
266266
vad_options=None,
267267
model : Optional[WhisperModel] = None,
268268
task="transcribe",
@@ -339,8 +339,8 @@ def load_model(whisper_arch,
339339
if vad_options is not None:
340340
default_vad_options.update(vad_options)
341341

342-
if vad_model is not None:
343-
vad_model = vad_model
342+
if vad_model_fp is not None:
343+
vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options, model_fp=vad_model_fp)
344344
else:
345345
vad_model = load_vad_model(torch.device(device), use_auth_token=None, **default_vad_options)
346346

0 commit comments

Comments
 (0)