From 445a4c69bbffebf059debebd291855a7915abc8c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20BREDIN?= Date: Wed, 11 Sep 2024 09:12:44 +0200 Subject: [PATCH] fix: fix support for CoreML export --- pyannote/audio/models/embedding/xvector.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pyannote/audio/models/embedding/xvector.py b/pyannote/audio/models/embedding/xvector.py index 3161876e3..55490bf47 100644 --- a/pyannote/audio/models/embedding/xvector.py +++ b/pyannote/audio/models/embedding/xvector.py @@ -342,8 +342,11 @@ def forward( Batch of weights with shape (batch, frame). """ - outputs = self.sincnet(waveforms).squeeze(dim=1) + # outputs = self.sincnet(waveforms).squeeze(dim=1) + outputs = self.sincnet(waveforms) + for tdnn in self.tdnns: outputs = tdnn(outputs) + outputs = self.stats_pool(outputs, weights=weights) return self.embedding(outputs)