haoneng.lhn 3 лет назад
Родитель
Сommit
bb5f0cfc3b
1 измененных файлов с 0 добавлено и 2 удалено
  1. 0 2
      funasr/models/predictor/cif.py

+ 0 - 2
funasr/models/predictor/cif.py

@@ -243,8 +243,6 @@ class CifPredictorV2(nn.Module):
                 pre_alphas_length = cache["cif_alphas"].size(-1)
                 pre_alphas_length = cache["cif_alphas"].size(-1)
                 mask_chunk_peak_predictor[:, :pre_alphas_length] = 1.0
                 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
             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:
         if mask_chunk_peak_predictor is not None:
             cif_peak = cif_peak * mask_chunk_peak_predictor.squeeze(-1)
             cif_peak = cif_peak * mask_chunk_peak_predictor.squeeze(-1)