| funasr/models/predictor/cif.py | ●●●●● 补丁 | 查看 | 原始文档 | 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)