liugz18
2024-06-12 648c03fdbba045acf8e25c5e6f7df0f27937c3cf
Update auto_model.py

修复空字串进入speaker model时报raw_text变量不存在的bug
1个文件已修改
5 ■■■■■ 已修改文件
funasr/auto/auto_model.py 5 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/auto/auto_model.py
@@ -469,10 +469,11 @@
            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"] = ""
                        result["raw_text"] = raw_text = ""
                else:
                    deep_update(self.punc_kwargs, cfg)
                    punc_res = self.inference(
@@ -482,8 +483,6 @@
                    if return_raw_text:
                        result["raw_text"] = raw_text
                    result["text"] = punc_res[0]["text"]
            else:
                raw_text = None
            # speaker embedding cluster after resorted
            if self.spk_model is not None and kwargs.get("return_spk_res", True):