Merge pull request #289 from Tomiinek/patch-1

Fixing ONNX init
snakers4 · Feb 28, 2023 · 4f57fae · 4f57fae
2 parents 262bcb4 + 085d76f
commit 4f57fae
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/utils_vad.py b/utils_vad.py
@@ -13,12 +13,15 @@ def __init__(self, path, force_onnx_cpu=False):
         import numpy as np
         global np
         import onnxruntime
+
+        opts = onnxruntime.SessionOptions()
+        opts.inter_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'])
+            self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts)
         else:
-            self.session = onnxruntime.InferenceSession(path)
-        self.session.intra_op_num_threads = 1
-        self.session.inter_op_num_threads = 1
+            self.session = onnxruntime.InferenceSession(path, sess_options=opts)
 
         self.reset_states()
         self.sample_rates = [8000, 16000]