mirror of
https://github.com/snakers4/silero-vad.git
synced 2026-02-04 17:39:22 +08:00
v4.0stable force_onnx_cpu fx
This commit is contained in:
@@ -34,7 +34,7 @@ def silero_vad(onnx=False, force_onnx_cpu=False):
|
|||||||
|
|
||||||
model_dir = os.path.join(os.path.dirname(__file__), 'files')
|
model_dir = os.path.join(os.path.dirname(__file__), 'files')
|
||||||
if onnx:
|
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:
|
else:
|
||||||
model = init_jit_model(os.path.join(model_dir, 'silero_vad.jit'))
|
model = init_jit_model(os.path.join(model_dir, 'silero_vad.jit'))
|
||||||
utils = (get_speech_timestamps,
|
utils = (get_speech_timestamps,
|
||||||
|
|||||||
Reference in New Issue
Block a user