Преглед изворни кода

fix paraformer online fbank cache bug

haoneng.lhn пре 2 година
родитељ
комит
47bd60924c
1 измењених фајлова са 3 додато и 0 уклоњено
  1. 3 0
      funasr/bin/asr_infer.py

+ 3 - 0
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()