diff --git a/whisperx/vad.py b/whisperx/vad.py index 0f15630..933d270 100644 --- a/whisperx/vad.py +++ b/whisperx/vad.py @@ -16,10 +16,11 @@ from typing import List, Tuple, Optional VAD_SEGMENTATION_URL = "https://whisperx.s3.eu-west-2.amazonaws.com/model_weights/segmentation/0b5b3216d60a2d32fc086b47ea8c67589aaeb26b7e07fcbe620d6d0b83e209ea/pytorch_model.bin" -def load_vad_model(device, vad_onset, vad_offset, use_auth_token=None): +def load_vad_model(device, vad_onset, vad_offset, use_auth_token=None, model_fp=None): model_dir = torch.hub._get_torch_home() os.makedirs(model_dir, exist_ok = True) - model_fp = os.path.join(model_dir, "whisperx-vad-segmentation.bin") + if model_fp is None: + model_fp = os.path.join(model_dir, "whisperx-vad-segmentation.bin") if os.path.exists(model_fp) and not os.path.isfile(model_fp): raise RuntimeError(f"{model_fp} exists and is not a regular file")