From 5da8a6b169e91b8a439530d8e57a3af46d33c8c2 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 31 Oct 2025 16:50:00 +0200
Subject: [PATCH] Fix MultiTalk on some models

---
 multitalk/multitalk.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/multitalk/multitalk.py b/multitalk/multitalk.py
index 220bb98..d017278 100644
--- a/multitalk/multitalk.py
+++ b/multitalk/multitalk.py
@@ -198,7 +198,7 @@ class AudioProjModel(nn.Module):
         context_tokens = self.proj3(audio_embeds_c).reshape(batch_size_c*N_t, self.context_tokens, self.output_dim)
 
         # normalization and reshape
-        context_tokens = self.norm(context_tokens)
+        context_tokens = self.norm(context_tokens.to(self.norm.weight.dtype)).to(context_tokens.dtype)
         context_tokens = rearrange(context_tokens, "(bz f) m c -> bz f m c", f=video_length)
 
         return context_tokens