| funasr/models/frontend/wav_frontend.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/models/frontend/wav_frontend.py
@@ -430,7 +430,7 @@ 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 @@ def cache_reset(self): self.reserve_waveforms = None self.input_cache = None self.lfr_splice_cache = [] self.lfr_splice_cache = []