Fix issue on old torch.

This commit is contained in:
comfyanonymous 2025-09-10 00:20:14 -04:00 committed by GitHub
parent 206595f854
commit d47eaac8b9
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,