fix small bug

badc5517 · Patrick von Platen · f448360b · badc5517
Commit badc5517 authored Jul 15, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/unet_unconditional.py src/diffusers/models/unet_unconditional.py +1 -1

No files found.
--- 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)