From cf252a8592c651648c781d0d6d6f1bc1a2f1c217 Mon Sep 17 00:00:00 2001 From: Max Bain Date: Fri, 14 Apr 2023 15:02:58 +0100 Subject: [PATCH] allow custom path for vad model --- whisperx/vad.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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")