From ee1b0ec605fc0c077867dfb6e1cbb65614eb1347 Mon Sep 17 00:00:00 2001 From: speech_asr Date: Tue, 14 Mar 2023 16:37:36 +0800 Subject: [PATCH] update --- funasr/models/frontend/wav_frontend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 = []