From 757d20b3e89401fc25817fb3de74c54dc9c6939f Mon Sep 17 00:00:00 2001
From: Wu Can <140670863+Hi99s@users.noreply.github.com>
Date: 星期一, 21 十月 2024 13:30:04 +0800
Subject: [PATCH] Fix typo (#2158)
---
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