From f6de1afc3febd680b41ba4b16d643cb3b897c091 Mon Sep 17 00:00:00 2001 From: Edna <88869424+Ednaordinary@users.noreply.github.com> Date: Wed, 11 Jun 2025 19:54:27 -0600 Subject: [PATCH] update --- src/diffusers/models/embeddings.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py index dc39480b65..8d3f7cbbe3 100644 --- a/src/diffusers/models/embeddings.py +++ b/src/diffusers/models/embeddings.py @@ -1665,8 +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 + return input_vec.to(dtype=timestep.dtype) class CogView3CombinedTimestepSizeEmbeddings(nn.Module): def __init__(self, embedding_dim: int, condition_dim: int, pooled_projection_dim: int, timesteps_dim: int = 256):