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

View File

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