Kyle He
2025-08-14 82a07e2f6ec60aa25a3931e9ee0d99ead642484a
fix: funasr_onnx pass run_options to ort (#2632)

1个文件已修改
4 ■■■■ 已修改文件
runtime/python/onnxruntime/funasr_onnx/utils/utils.py 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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