From 9c1bf379603f3f22466fde1a0818d2edda95ea6d Mon Sep 17 00:00:00 2001
From: Quentin Fuxa <quentin.fuxa@gmail.com>
Date: Fri, 5 Sep 2025 16:34:13 +0200
Subject: [PATCH] fixes #197

---
 whisperlivekit/diarization/sortformer_backend.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/whisperlivekit/diarization/sortformer_backend.py b/whisperlivekit/diarization/sortformer_backend.py
index dd35cfd..3072eae 100644
--- 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 *