Hui Wang
2024-04-09 eef1c2a5b885ebc045373a98cb5d3f66019c65b7
funasr/models/contextual_paraformer/model.py
@@ -94,10 +94,8 @@
                text: (Batch, Length)
                text_lengths: (Batch,)
        """
        if len(text_lengths.size()) > 1:
            text_lengths = text_lengths[:, 0]
        if len(speech_lengths.size()) > 1:
            speech_lengths = speech_lengths[:, 0]
        text_lengths = text_lengths.squeeze()
        speech_lengths = speech_lengths.squeeze()
        batch_size = speech.shape[0]