fixes #197

2025-09-05 16:34:13 +02:00 · 2025-09-05 16:34:13 +02:00 · 9c1bf37960
commit 9c1bf37960
parent f46528471b
1 changed files with 1 additions and 0 deletions
--- a/whisperlivekit/diarization/sortformer_backend.py
+++ b/whisperlivekit/diarization/sortformer_backend.py
@ -106,6 +106,7 @@ class SortformerDiarizationOnline:
            features=128,
            pad_to=0
        )
+        self.audio2mel.to(self.diar_model.device)
        
        self.chunk_duration_seconds = (
            self.diar_model.sortformer_modules.chunk_len *