雾聪
2024-09-26 cd4c6485efd9c1ff306f1708cbba6d2ea582250b
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: