cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #52 from makaveli10/vad_warnings

Browse Source 
Add `force_onnx_cpu` arg, thanks Shubh for reporting the issue and providing a fix.
This commit is contained in:
Marcus Edel
2023-09-28 15:01:43 -04:00
committed by GitHub
parent 44e0db1634 ad37d8c63c
commit b5f9f1ed08
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/vad.py
+6 -2
View File
@@ -9,7 +9,7 @@ import onnxruntime
class VoiceActivityDetection(): class VoiceActivityDetection():
def __init__(self): def __init__(self, force_onnx_cpu=True):
path = self.download() path = self.download()
opts = onnxruntime.SessionOptions() opts = onnxruntime.SessionOptions()
opts.log_severity_level = 3 opts.log_severity_level = 3
@@ -17,7 +17,11 @@ class VoiceActivityDetection():
opts.inter_op_num_threads = 1 opts.inter_op_num_threads = 1
opts.intra_op_num_threads = 1 opts.intra_op_num_threads = 1
self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts) if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers():
self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts)
else:
self.session = onnxruntime.InferenceSession(path, providers=['CUDAExecutionProvider'], sess_options=opts)
self.reset_states() self.reset_states()
self.sample_rates = [8000, 16000] self.sample_rates = [8000, 16000]