AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

undo don't change dtype

Browse Source
This commit is contained in:
Edna
2025-06-12 00:46:41 -06:00
committed by GitHub
parent b0cf6803a7
commit 0c5eb44701
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/embeddings.py
View File

@@ -1658,7 +1658,7 @@ class CombinedTimestepTextProjChromaEmbeddings(nn.Module):
torch.cat([timesteps_proj, guidance_proj], dim=1).unsqueeze(1).repeat(1, mod_index_length, 1)
)
input_vec = torch.cat([timestep_guidance, mod_proj.unsqueeze(0)], dim=-1)
return input_vec
return input_vec.to(timestep.dtype)
class CogView3CombinedTimestepSizeEmbeddings(nn.Module):
def __init__(self, embedding_dim: int, condition_dim: int, pooled_projection_dim: int, timesteps_dim: int = 256):