Fix issue on old torch. (#9791)

This commit is contained in:
comfyanonymous 2025-09-09 21:23:47 -07:00 committed by GitHub
parent 206595f854
commit 5c33872e2f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -426,7 +426,7 @@ class HunYuanDiTBlock(nn.Module):
text_states_dim=1024,
qk_norm=False,
norm_layer=nn.LayerNorm,
qk_norm_layer=nn.RMSNorm,
qk_norm_layer=True,
qkv_bias=True,
skip_connection=True,
timested_modulate=False,