From 719e6facf9fe9fdb8d096b76b59844dae3274d6e Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 13 May 2026 21:58:16 +0300
Subject: [PATCH] Update modules.py

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

diff --git a/comfy/ldm/moge/modules.py b/comfy/ldm/moge/modules.py
index ff7b0878a..235a59212 100644
--- a/comfy/ldm/moge/modules.py
+++ b/comfy/ldm/moge/modules.py
@@ -36,8 +36,8 @@ class ResidualConvBlock(nn.Module):
         super().__init__()
         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()
-        hidden_norm_layer = (operations.GroupNorm(max(hidden_channels // 32, 1), hidden_channels)
+        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, dtype=dtype, device=device)
                              if hidden_norm == "group_norm" else nn.Identity())
 
         self.layers = nn.Sequential(