From 8745733b5471f9e0d59114d3c99d9041a8b831c8 Mon Sep 17 00:00:00 2001 From: lyblsgo Date: Wed, 25 Sep 2024 20:24:42 +0800 Subject: [PATCH] paraformer_bin bugfix --- runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py b/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py index a60b6d64d..4f35fcc86 100644 --- 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