凌匀
2023-03-13 2af58ee95358b0871dfc18631ec7d870b5cd53f3
funasr/bin/vad_inference_online.py
@@ -96,7 +96,7 @@
            }
            # 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