speech_asr 3 лет назад
Родитель
Сommit
ee1b0ec605
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      funasr/models/frontend/wav_frontend.py

+ 2 - 2
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 = []
+        self.lfr_splice_cache = []