Kaynağa Gözat

Merge pull request #221 from alibaba-damo-academy/dev_zly

update vad_inference_online.py
hnluo 3 yıl önce
ebeveyn
işleme
08d6f5cc1c
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      funasr/bin/vad_inference_online.py

+ 1 - 1
funasr/bin/vad_inference_online.py

@@ -96,7 +96,7 @@ class Speech2VadSegmentOnline(Speech2VadSegment):
             }
             # a. To device
             batch = to_device(batch, device=self.device)
-            segments, in_cache = self.vad_model(**batch)
+            segments, in_cache = self.vad_model.forward_online(**batch)
             # in_cache.update(batch['in_cache'])
             # in_cache = {key: value for key, value in batch['in_cache'].items()}
         return fbanks, segments, in_cache