From a76f15c78527edf47a68dd67e2c4d894bc252398 Mon Sep 17 00:00:00 2001
From: Djraemon <154407769+Djraemon@users.noreply.github.com>
Date: 星期一, 21 十月 2024 13:30:45 +0800
Subject: [PATCH] Fix audio format (#2159)
---
runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
index a60b6d6..4f35fcc 100644
--- a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
+++ b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
@@ -322,6 +322,10 @@
self.pred_bias = config["model_conf"]["predictor_bias"]
else:
self.pred_bias = 0
+ if "lang" in config:
+ self.language = config["lang"]
+ else:
+ self.language = None
def __call__(
self, wav_content: Union[str, np.ndarray, List[str]], hotwords: str, **kwargs
--
Gitblit v1.9.1