chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Added Vad custom option

Browse Source
This commit is contained in:
Full Name
2024-01-01 14:56:51 +05:30
parent f8cc46c6f7
commit 6bb2f1cd48
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisperx/asr.py
View File

@ -262,6 +262,7 @@ def load_model(whisper_arch,
compute_type="float16", compute_type="float16",
asr_options=None, asr_options=None,
language : Optional[str] = None, language : Optional[str] = None,
vad_model=None,
vad_options=None, vad_options=None,
model : Optional[WhisperModel] = None, model : Optional[WhisperModel] = None,
task="transcribe", task="transcribe",
@ -337,7 +338,10 @@ def load_model(whisper_arch,
if vad_options is not None: if vad_options is not None:
default_vad_options.update(vad_options) 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( return FasterWhisperPipeline(
model=model, model=model,