From 8c6d1642f5fbf1d55edb324e35e9fa6e89da25a1 Mon Sep 17 00:00:00 2001
From: Xuning Tan <me@xuning.eu.org>
Date: 星期三, 01 十月 2025 14:45:45 +0800
Subject: [PATCH] fix: correct the deepspeed_config path reference in the finetuning script (#2680)
---
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