mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-14 21:45:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lowvram issue with hunyuan image vae. (#9794)

Browse Source
This commit is contained in:
comfyanonymous
2025-09-09 23:15:34 -07:00
committed by GitHub
parent 70fc0425b3
commit 543888d3d8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/hunyuan_video/vae.py
View File

@@ -65,7 +65,7 @@ class Encoder(nn.Module):
self.mid.attn_1 = AttnBlock(ch, conv_op=ops.Conv2d) 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.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) self.conv_out = ops.Conv2d(ch, z_channels << 1, 3, 1, 1)
def forward(self, x): def forward(self, x):
@@ -120,7 +120,7 @@ class Decoder(nn.Module):
ch = nxt ch = nxt
self.up.append(stage) 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) self.conv_out = ops.Conv2d(ch, out_channels, 3, 1, 1)
def forward(self, z): def forward(self, z):