paraformer_bin bugfix

2025-09-15 14:48:36 +08:00 · 2024-09-25 20:24:42 +08:00 · 2024-09-25 20:24:42 +08:00 · 8745733b54
commit 8745733b54
parent fc00476a25
1 changed files with 4 additions and 0 deletions
--- a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
+++ b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
@ -322,6 +322,10 @@ class ContextualParaformer(Paraformer):
            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