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

Merge pull request #255 from snakers4/adamnsandle

Browse Source 
Adamnsandle
This commit is contained in:
Dimitrii Voronin
2022-10-28 13:33:18 +03:00
committed by GitHub
parent c7eab751b5 888970e77d
commit 51b5245410
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
hubconf.py
View File

@@ -1,6 +1,7 @@
dependencies = ['torch', 'torchaudio']
import torch
import json
import warnings
from utils_vad import (init_jit_model,
get_speech_timestamps,
get_number_ts,
@@ -16,7 +17,7 @@ from utils_vad import (init_jit_model,
def versiontuple(v):
return tuple(map(int, (v.split("."))))
return tuple(map(int, (v.split('+')[0].split("."))))
def silero_vad(onnx=False, force_onnx_cpu=False):
@@ -25,11 +26,11 @@ def silero_vad(onnx=False, force_onnx_cpu=False):
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)')
if not onnx:
installed_version = torch.__version__
supported_version = '1.12.0'
if versiontuple(installed_version) < versiontuple(supported_version):
warnings.warn(f'Please install torch {supported_version} or greater ({installed_version} installed)')
hub_dir = torch.hub.get_dir()
if onnx: