From 543888d3d84a6ec4c4273838d5179845840e3226 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 9 Sep 2025 23:15:34 -0700
Subject: [PATCH] Fix lowvram issue with hunyuan image vae. (#9794)

---
 comfy/ldm/hunyuan_video/vae.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy/ldm/hunyuan_video/vae.py b/comfy/ldm/hunyuan_video/vae.py
index 8d406089b..40c12b183 100644
--- a/comfy/ldm/hunyuan_video/vae.py
+++ b/comfy/ldm/hunyuan_video/vae.py
@@ -65,7 +65,7 @@ class Encoder(nn.Module):
         self.mid.attn_1 = AttnBlock(ch, conv_op=ops.Conv2d)
         self.mid.block_2 = ResnetBlock(in_channels=ch, out_channels=ch, temb_channels=0, conv_op=ops.Conv2d)
 
-        self.norm_out = nn.GroupNorm(32, ch, 1e-6, True)
+        self.norm_out = ops.GroupNorm(32, ch, 1e-6, True)
         self.conv_out = ops.Conv2d(ch, z_channels << 1, 3, 1, 1)
 
     def forward(self, x):
@@ -120,7 +120,7 @@ class Decoder(nn.Module):
                 ch = nxt
             self.up.append(stage)
 
-        self.norm_out = nn.GroupNorm(32, ch, 1e-6, True)
+        self.norm_out = ops.GroupNorm(32, ch, 1e-6, True)
         self.conv_out = ops.Conv2d(ch, out_channels, 3, 1, 1)
 
     def forward(self, z):