wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-16 03:57:27 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Update modules.py

Browse Source
This commit is contained in:
kijai 2026-05-13 21:58:16 +03:00
parent 6617e76b1c
commit 719e6facf9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/moge/modules.py
View File

@ -36,8 +36,8 @@ class ResidualConvBlock(nn.Module):
super().__init__() super().__init__()
hidden_channels = hidden_channels if hidden_channels is not None else channels hidden_channels = hidden_channels if hidden_channels is not None else channels
in_norm_layer = operations.GroupNorm(1, channels) if in_norm == "layer_norm" else nn.Identity() in_norm_layer = operations.GroupNorm(1, channels, dtype=dtype, device=device) if in_norm == "layer_norm" else nn.Identity()
hidden_norm_layer = (operations.GroupNorm(max(hidden_channels // 32, 1), hidden_channels) hidden_norm_layer = (operations.GroupNorm(max(hidden_channels // 32, 1), hidden_channels, dtype=dtype, device=device)
if hidden_norm == "group_norm" else nn.Identity()) if hidden_norm == "group_norm" else nn.Identity())
self.layers = nn.Sequential( self.layers = nn.Sequential(