语帆
2024-02-22 c1c337ef9a6916d9fb12898983f54b5f3630ff0a
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: