v4.0stable force_onnx_cpu fx

This commit is contained in:
adamnsandle
2024-07-08 10:16:52 +00:00
parent ac128b3c55
commit 7a176cc294

View File

@@ -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,