From badc5517ff1ec433c1f417fcdd90b467e650cedb Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 15 Jul 2022 14:03:08 +0000 Subject: [PATCH] fix small bug --- src/diffusers/models/unet_unconditional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/unet_unconditional.py b/src/diffusers/models/unet_unconditional.py index ad738a2737..1a61891c4f 100644 --- a/src/diffusers/models/unet_unconditional.py +++ b/src/diffusers/models/unet_unconditional.py @@ -201,7 +201,7 @@ class UNetUnconditionalModel(ModelMixin, ConfigMixin): prev_output_channel = output_channel # out - self.conv_norm_out = nn.GroupNorm(num_channels=block_channels[0], num_groups=32, eps=1e-5) + self.conv_norm_out = nn.GroupNorm(num_channels=block_channels[0], num_groups=32, eps=resnet_eps) self.conv_act = nn.SiLU() self.conv_out = nn.Conv2d(block_channels[0], out_channels, 3, padding=1)