From 82a07e2f6ec60aa25a3931e9ee0d99ead642484a Mon Sep 17 00:00:00 2001
From: Kyle He <599316527@users.noreply.github.com>
Date: 星期四, 14 八月 2025 16:07:54 +0800
Subject: [PATCH] fix: funasr_onnx pass run_options to ort (#2632)
---
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