haoneng.lhn
2023-03-29 bb5f0cfc3b6fb71d24cf93d1a47927dc06b4c716
update
1个文件已修改
2 ■■■■■ 已修改文件
funasr/models/predictor/cif.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/predictor/cif.py
@@ -243,8 +243,6 @@
                pre_alphas_length = cache["cif_alphas"].size(-1)
                mask_chunk_peak_predictor[:, :pre_alphas_length] = 1.0
            mask_chunk_peak_predictor[:, pre_alphas_length + cache["pad_left"]:pre_alphas_length + cache["stride"] + cache["pad_left"]] = 1.0
            #if cache["is_final"]:
            #    mask_chunk_peak_predictor[:, -1] = 1.0
            
        if mask_chunk_peak_predictor is not None:
            cif_peak = cif_peak * mask_chunk_peak_predictor.squeeze(-1)