From 5c33872e2f355e51adf212d5b5c83815b7fe77b0 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 9 Sep 2025 21:23:47 -0700
Subject: [PATCH] Fix issue on old torch. (#9791)

---
 comfy/ldm/hunyuan3dv2_1/hunyuandit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/ldm/hunyuan3dv2_1/hunyuandit.py b/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
index ca1a83001..d48d9d642 100644
--- 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,