diff --git a/funasr/models/frontend/wav_frontend.py b/funasr/models/frontend/wav_frontend.py index 0bf5ce16d..445efca24 100644 --- a/funasr/models/frontend/wav_frontend.py +++ b/funasr/models/frontend/wav_frontend.py @@ -430,7 +430,7 @@ class WavFrontendOnline(AbsFrontend): else: if is_final: self.waveforms = waveforms if self.reserve_waveforms is None else self.reserve_waveforms - feats = torch.stack(self.lfr_splice_cache) + feats = torch.stack(self.lfr_splice_cache) feats_lengths = torch.zeros(batch_size, dtype=torch.int) + feats.shape[1] feats, feats_lengths, _ = self.forward_lfr_cmvn(feats, feats_lengths, is_final) if is_final: @@ -443,4 +443,4 @@ class WavFrontendOnline(AbsFrontend): def cache_reset(self): self.reserve_waveforms = None self.input_cache = None - self.lfr_splice_cache = [] \ No newline at end of file + self.lfr_splice_cache = []