Fix MultiTalk on some models

2025-10-31 16:50:00 +02:00
parent ce6e7b501d
commit 5da8a6b169
1 changed files with 1 additions and 1 deletions
@@ -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