mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-29 18:23:59 +08:00
0e43748fdc
* 🦄 refactor(web): 修改zhikongtaiweb到web * 🦄 refactor: 重写前端 路由守护尚未写完 * 🦄 refactor: 标准化路由 * update:前端重写，保留后端 * update:添加前端代码 * update:pip转成poetry启动 * update:增加mem0ai包依赖 * update:文档增加mem0ai的描述 * feat: play online mp3 (#181) Co-authored-by: 欣南科技 <huangrongzhuang@xin-nan.com> * 修改前端代码 * update:调整项目目录 * update:优化 * update:配置文件去除8002端口 * update:增加开发说明 * update:更新开发协议 --------- Co-authored-by: kalicyh <34980061+kaliCYH@users.noreply.github.com> Co-authored-by: hrz <1710360675@qq.com> Co-authored-by: freshlife001 <talent@mises.site> Co-authored-by: CGD <3030332422@qq.com>
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from .utils_vad import init_jit_model, OnnxWrapper
|
|
import torch
|
|
torch.set_num_threads(1)
|
|
|
|
|
|
def load_silero_vad(onnx=False, opset_version=16):
|
|
available_ops = [15, 16]
|
|
if onnx and opset_version not in available_ops:
|
|
raise Exception(f'Available ONNX opset_version: {available_ops}')
|
|
|
|
if onnx:
|
|
if opset_version == 16:
|
|
model_name = 'silero_vad.onnx'
|
|
else:
|
|
model_name = f'silero_vad_16k_op{opset_version}.onnx'
|
|
else:
|
|
model_name = 'silero_vad.jit'
|
|
package_path = "silero_vad.data"
|
|
|
|
try:
|
|
import importlib_resources as impresources
|
|
model_file_path = str(impresources.files(package_path).joinpath(model_name))
|
|
except:
|
|
from importlib import resources as impresources
|
|
try:
|
|
with impresources.path(package_path, model_name) as f:
|
|
model_file_path = f
|
|
except:
|
|
model_file_path = str(impresources.files(package_path).joinpath(model_name))
|
|
|
|
if onnx:
|
|
model = OnnxWrapper(model_file_path, force_onnx_cpu=True)
|
|
else:
|
|
model = init_jit_model(model_file_path)
|
|
|
|
return model
|