o
    ã¥µi{  ã                   @   s0   d dl mZmZ ddlZe d ¡ ddd„ZdS )	é   )Úinit_jit_modelÚOnnxWrapperé    NFé   c           	      C   sö   ddg}| r||vrt d|› ƒ‚| r!|dkrd}n	d|› d}nd}d}zd	d l}t| |¡ |¡ƒ}W n3   d	d
lm} z| ||¡
}|}W d   ƒ n1 sSw   Y  W n   t| |¡ |¡ƒ}Y Y | rutt|ƒdd}|S t	|ƒ}|S )Né   r   zAvailable ONNX opset_version: zsilero_vad.onnxÚsilero_vad_16k_opz.onnxzsilero_vad.jitzsilero_vad.datar   )Ú	resourcesT)Úforce_onnx_cpu)
Ú	ExceptionÚimportlib_resourcesÚstrÚfilesÚjoinpathÚ	importlibr   Úpathr   r   )	ÚonnxÚopset_versionÚavailable_opsÚ
model_nameÚpackage_pathÚimpresourcesÚmodel_file_pathÚfÚmodel© r   úD/home/ubuntu/.local/lib/python3.10/site-packages/silero_vad/model.pyÚload_silero_vad   s6   ÿ€€þr   )Fr   )Ú	utils_vadr   r   ÚtorchÚset_num_threadsr   r   r   r   r   Ú<module>   s    
