From f783f38883f6f9c04c6ccb0a5bb630cc76c07e98 Mon Sep 17 00:00:00 2001
From: Edna <88869424+Ednaordinary@users.noreply.github.com>
Date: Wed, 11 Jun 2025 19:52:43 -0600
Subject: [PATCH] ensure correct dtype for chroma embeddings

---
 src/diffusers/models/embeddings.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py
index a319992675..dc39480b65 100644
--- a/src/diffusers/models/embeddings.py
+++ b/src/diffusers/models/embeddings.py
@@ -1665,6 +1665,7 @@ class CombinedTimestepTextProjChromaEmbeddings(nn.Module):
             torch.cat([timesteps_proj, guidance_proj], dim=1).repeat(1, mod_index_length, 1)
         )
         input_vec = torch.cat([timestep_guidance, mod_proj], dim=-1)
+        input_vec.to(dtype=timestep.dtype)
         return input_vec
 
 class CogView3CombinedTimestepSizeEmbeddings(nn.Module):