夜雨飘零
2023-12-19 53fccccb24d15d788919d91c8c2b06a115ddacf3
funasr/models/frontend/wav_frontend.py
@@ -116,7 +116,7 @@
    def forward(
            self,
            input: torch.Tensor,
            input_lengths: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
            input_lengths) -> Tuple[torch.Tensor, torch.Tensor]:
        batch_size = input.size(0)
        feats = []
        feats_lens = []
@@ -146,7 +146,7 @@
        feats_lens = torch.as_tensor(feats_lens)
        if batch_size == 1:
            feats_pad = feats[None, :, :]
            feats_pad = feats[0][None, :, :]
        else:
            feats_pad = pad_sequence(feats,
                                     batch_first=True,