diff --git a/funasr/bin/vad_inference_online.py b/funasr/bin/vad_inference_online.py index cee1929fb..b33d1a057 100644 --- a/funasr/bin/vad_inference_online.py +++ b/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