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,