Просмотр исходного кода

Update asr_inference_paraformer_streaming.py

hnluo 2 лет назад
Родитель
Сommit
6abf68388c
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      funasr/bin/asr_inference_paraformer_streaming.py

+ 3 - 3
funasr/bin/asr_inference_paraformer_streaming.py

@@ -202,9 +202,9 @@ class Speech2Text:
         assert check_argument_types()
         assert check_argument_types()
         results = []
         results = []
         cache_en = cache["encoder"]
         cache_en = cache["encoder"]
-        if speech.shape[1] < 16 * 60 and cache["is_final"]:
-            cache["last_chunk"] = True
-            feats = cache["feats"]
+        if speech.shape[1] < 16 * 60 and cache_en["is_final"]:
+            cache_en["last_chunk"] = True
+            feats = cache_en["feats"]
             feats_len = torch.tensor([feats.shape[1]])
             feats_len = torch.tensor([feats.shape[1]])
         else:
         else:
             if self.frontend is not None:
             if self.frontend is not None: