From eb1574b813e230b156fc09eaaf03227b1b0b4134 Mon Sep 17 00:00:00 2001
From: weilikai <jasper@talkus.fun>
Date: 星期六, 20 九月 2025 22:41:05 +0800
Subject: [PATCH] fix: support loading .pcm (16k 1c 16bit) audio files in load_utils.py (#2667) (#2668)
---
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