zhifu gao
2024-02-06 d92cd5ae037ae85ab9730499d99e5c1bd475eed2
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: