From 915dd3d639b8333a52e001af095f87c5b7f1e0ac Mon Sep 17 00:00:00 2001 From: adamnsandle Date: Mon, 8 Jul 2024 10:16:52 +0000 Subject: [PATCH] v4.0stable force_onnx_cpu fx --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 9eab444..c058431 100644 --- a/hubconf.py +++ b/hubconf.py @@ -34,7 +34,7 @@ def silero_vad(onnx=False, force_onnx_cpu=False): model_dir = os.path.join(os.path.dirname(__file__), 'files') if onnx: - model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx')) + model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx'), force_onnx_cpu) else: model = init_jit_model(os.path.join(model_dir, 'silero_vad.jit')) utils = (get_speech_timestamps,