Fix issue on old torch. (#9791)

2025-09-09 21:23:47 -07:00 · 2025-09-09 21:23:47 -07:00 · 5c33872e2f
parent 206595f854
commit 5c33872e2f
1 changed files with 1 additions and 1 deletions
--- a/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
+++ b/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
@ -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,