From 47bd60924c47baefd37f9f12866ec0d24055bc62 Mon Sep 17 00:00:00 2001 From: "haoneng.lhn" Date: Thu, 18 May 2023 19:52:10 +0800 Subject: [PATCH] fix paraformer online fbank cache bug --- funasr/bin/asr_infer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/funasr/bin/asr_infer.py b/funasr/bin/asr_infer.py index d9d413b3a..59462711f 100644 --- a/funasr/bin/asr_infer.py +++ b/funasr/bin/asr_infer.py @@ -749,10 +749,13 @@ class Speech2TextParaformerOnline: feats = cache_en["feats"] feats_len = torch.tensor([feats.shape[1]]) self.asr_model.frontend = None + self.frontend.cache_reset() results = self.infer(feats, feats_len, cache) return results else: if self.frontend is not None: + if cache_en["start_idx"] == 0: + self.frontend.cache_reset() feats, feats_len = self.frontend.forward(speech, speech_lengths, cache_en["is_final"]) feats = to_device(feats, device=self.device) feats_len = feats_len.int()