游雁
2024-10-29 7699a35d2cb08f4f26c00aed21e105dd5870c538
funasr/models/transformer/model.py
@@ -145,8 +145,6 @@
                text: (Batch, Length)
                text_lengths: (Batch,)
        """
        # import pdb;
        # pdb.set_trace()
        if len(text_lengths.size()) > 1:
            text_lengths = text_lengths[:, 0]
        if len(speech_lengths.size()) > 1: