From 73b2c9e3d09b189c9addb7de18c8414a07675e7c Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 30 Jul 2024 01:32:19 -0400 Subject: [PATCH] Pin SileroVAD to v4.0 Signed-off-by: makaveli10 --- docker/scripts/setup-whisperfusion.sh | 2 +- whisper_live/vad.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/scripts/setup-whisperfusion.sh b/docker/scripts/setup-whisperfusion.sh index 75e3306..c32dbd9 100755 --- a/docker/scripts/setup-whisperfusion.sh +++ b/docker/scripts/setup-whisperfusion.sh @@ -19,6 +19,6 @@ huggingface-cli download charactr/vocos-encodec-24khz mkdir -p /root/.cache/torch/hub/checkpoints/ curl -L -o /root/.cache/torch/hub/checkpoints/encodec_24khz-d7cc33bc.th https://dl.fbaipublicfiles.com/encodec/v0/encodec_24khz-d7cc33bc.th mkdir -p /root/.cache/whisper-live/ -curl -L -o /root/.cache/whisper-live/silero_vad.onnx https://github.com/snakers4/silero-vad/raw/master/files/silero_vad.onnx +curl -L -o /root/.cache/whisper-live/silero_vad.onnx https://github.com/snakers4/silero-vad/raw/v4.0/files/silero_vad.onnx python3 -c 'from transformers.utils.hub import move_cache; move_cache()' diff --git a/whisper_live/vad.py b/whisper_live/vad.py index ec93333..1287c46 100644 --- a/whisper_live/vad.py +++ b/whisper_live/vad.py @@ -98,7 +98,7 @@ def audio_forward(self, x, sr: int, num_samples: int = 512): return stacked.cpu() @staticmethod - def download(model_url="https://github.com/snakers4/silero-vad/raw/master/files/silero_vad.onnx"): + def download(model_url="https://github.com/snakers4/silero-vad/raw/v4.0/files/silero_vad.onnx"): target_dir = os.path.expanduser("~/.cache/whisper-live/") # Ensure the target directory exists