liugz18
2024-06-12 bac3ef2e250cbb06bc5815af188f165f68ed2066
Update auto_model.py

修复识别出空串后spk_model内变量未定义问题
1个文件已修改
6 ■■■■■ 已修改文件
funasr/auto/auto_model.py 6 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/auto/auto_model.py
@@ -467,14 +467,12 @@
                        else:
                            result[k] += restored_data[j][k]
            if not len(result["text"].strip()):
                continue
            return_raw_text = kwargs.get("return_raw_text", False)
            # step.3 compute punc model
            raw_text = None
            if self.punc_model is not None:
                if not len(result["text"].strip()):
                    if return_raw_text:
                        result["raw_text"] = raw_text = ""
                else:
                    deep_update(self.punc_kwargs, cfg)
                    punc_res = self.inference(
                        result["text"], model=self.punc_model, kwargs=self.punc_kwargs, **cfg