funasr/models/e2e_diar_eend_ola.py @@ -240,3 +240,6 @@ torch.float32) decisions = decisions[:, :n_speaker] return decisions def collect_feats(self, **batch: torch.Tensor) -> Dict[str, torch.Tensor]: pass
