fix huanyan

SunMarc · SunMarc · commit b4e4f3b78a42 · 2025-01-18T11:45:24.000+01:00
diff --git a/src/diffusers/models/transformers/hunyuan_transformer_2d.py b/src/diffusers/models/transformers/hunyuan_transformer_2d.py
@@ -278,7 +278,7 @@ def __init__(
         )
 
         self.text_embedding_padding = nn.Parameter(
-            torch.randn(text_len + text_len_t5, cross_attention_dim, dtype=torch.float32)
+            torch.randn(text_len + text_len_t5, cross_attention_dim)
         )
 
         self.pos_embed = PatchEmbed(

Original file line number	Diff line number	Diff line change
`@@ -278,7 +278,7 @@ def __init__(`
`278`	`278`	`)`
`279`	`279`
`280`	`280`	`self.text_embedding_padding = nn.Parameter(`
`281`		`- torch.randn(text_len + text_len_t5, cross_attention_dim, dtype=torch.float32)`
	`281`	`+ torch.randn(text_len + text_len_t5, cross_attention_dim)`
`282`	`282`	`)`
`283`	`283`
`284`	`284`	`self.pos_embed = PatchEmbed(`