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

allow only CPU Execution

Browse Source
This commit is contained in:
makaveli10
2023-09-27 16:17:20 +08:00
parent 0444cea0ba
commit 5fb6be41bb
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/vad.py
+2 -5
View File
@@ -9,7 +9,7 @@ import onnxruntime
class VoiceActivityDetection(): class VoiceActivityDetection():
def __init__(self, force_onnx_cpu=True): def __init__(self):
path = self.download() path = self.download()
opts = onnxruntime.SessionOptions() opts = onnxruntime.SessionOptions()
opts.log_severity_level = 3 opts.log_severity_level = 3
@@ -17,10 +17,7 @@ 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
if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers(): self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts)
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]