游雁
2024-02-20 f3c79b4968be8ffd4bedb40baff57d8e70384008
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: