游雁
2024-07-29 eb86f0fe6a7605c1e72fb8307d525ee8b8ebfb1e
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: