From 79040635dace9466a9f3fe20b5604b8e8e79f44f Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 30 Jul 2024 05:01:34 -0400 Subject: [PATCH] Remove unnecessary code. --- comfy/ldm/modules/diffusionmodules/openaimodel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/modules/diffusionmodules/openaimodel.py b/comfy/ldm/modules/diffusionmodules/openaimodel.py index ba8fc2c4..6535e899 100644 --- a/comfy/ldm/modules/diffusionmodules/openaimodel.py +++ b/comfy/ldm/modules/diffusionmodules/openaimodel.py @@ -809,7 +809,7 @@ class UNetModel(nn.Module): self.out = nn.Sequential( operations.GroupNorm(32, ch, dtype=self.dtype, device=device), nn.SiLU(), - zero_module(operations.conv_nd(dims, model_channels, out_channels, 3, padding=1, dtype=self.dtype, device=device)), + operations.conv_nd(dims, model_channels, out_channels, 3, padding=1, dtype=self.dtype, device=device), ) if self.predict_codebook_ids: self.id_predictor = nn.Sequential(