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

Merge pull request #654 from Swami-Abhinav/provide-custom-load-vad

Browse Source 
Added option to load Custom VAD model to load model method
This commit is contained in:
Max Bain
2024-01-01 17:38:30 +00:00
committed by GitHub
parent f8cc46c6f7 6bb2f1cd48
commit 06e30b2a25
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",
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,