From 5115a066c971b2380c98f9be168e14869053eda1 Mon Sep 17 00:00:00 2001
From: majic31 <majic31@163.com>
Date: 星期四, 14 八月 2025 16:08:44 +0800
Subject: [PATCH] fix #2587: Resolve VAD multithreading issue (#2613)
---
runtime/python/onnxruntime/funasr_onnx/utils/utils.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/runtime/python/onnxruntime/funasr_onnx/utils/utils.py b/runtime/python/onnxruntime/funasr_onnx/utils/utils.py
index c39d433..e176fb6 100644
--- a/runtime/python/onnxruntime/funasr_onnx/utils/utils.py
+++ b/runtime/python/onnxruntime/funasr_onnx/utils/utils.py
@@ -221,10 +221,10 @@
RuntimeWarning,
)
- def __call__(self, input_content: List[Union[np.ndarray, np.ndarray]]) -> np.ndarray:
+ def __call__(self, input_content: List[Union[np.ndarray, np.ndarray]], run_options = None) -> np.ndarray:
input_dict = dict(zip(self.get_input_names(), input_content))
try:
- return self.session.run(self.get_output_names(), input_dict)
+ return self.session.run(self.get_output_names(), input_dict, run_options)
except Exception as e:
raise ONNXRuntimeError("ONNXRuntime inferece failed.") from e
--
Gitblit v1.9.1