deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-05 01:49:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #253 from snakers4/adamnsandle

Browse Source 
add torch version check
This commit is contained in:
Dimitrii Voronin
2022-10-28 13:09:18 +03:00
committed by GitHub
parent 94c79d899d d1714a9ff7
commit c7eab751b5
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
hubconf.py
View File

@@ -15,11 +15,22 @@ from utils_vad import (init_jit_model,
OnnxWrapper) OnnxWrapper)
def versiontuple(v):
return tuple(map(int, (v.split("."))))
def silero_vad(onnx=False, force_onnx_cpu=False): def silero_vad(onnx=False, force_onnx_cpu=False):
"""Silero Voice Activity Detector """Silero Voice Activity Detector
Returns a model with a set of utils Returns a model with a set of utils
Please see https://github.com/snakers4/silero-vad for usage examples Please see https://github.com/snakers4/silero-vad for usage examples
""" """
installed_version = torch.__version__
supported_version = '1.12.0'
if versiontuple(installed_version) < versiontuple(supported_version):
raise Exception(f'Please install torch {supported_version} or greater ({installed_version} installed)')
hub_dir = torch.hub.get_dir() hub_dir = torch.hub.get_dir()
if onnx: if onnx:
model = OnnxWrapper(f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.onnx', force_onnx_cpu) model = OnnxWrapper(f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.onnx', force_onnx_cpu)