语帆
2024-03-04 1d7ba1be1ad824135698e8000386c1fd55268ae4
funasr/models/paraformer/cif_predictor.py
@@ -186,7 +186,7 @@
        alphas = alphas.squeeze(-1)
        mask = mask.squeeze(-1)
        if target_label_length is not None:
            target_length = target_label_length
            target_length = target_label_length.squeeze(-1)
        elif target_label is not None:
            target_length = (target_label != ignore_id).float().sum(-1)
        else: