From 293a9acc2ae91bcbf4ba6591b6f761c790d25097 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Thu, 26 Jan 2023 14:04:46 +0100 Subject: [PATCH] Fix downsample channels. Reference: https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/unet_2d_blocks.py#L749-L756 This is usually not a problem for Stable Diffusion models. However, some architectures such as https://huggingface.co/OFA-Sys/small-stable-diffusion-v0 wouldn't convert. --- python_coreml_stable_diffusion/unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python_coreml_stable_diffusion/unet.py b/python_coreml_stable_diffusion/unet.py index 581cfcd..f75c7f2 100644 --- a/python_coreml_stable_diffusion/unet.py +++ b/python_coreml_stable_diffusion/unet.py @@ -376,7 +376,7 @@ def __init__( self.resnets = nn.ModuleList(resnets) if add_downsample: - self.downsamplers = nn.ModuleList([Downsample2D(in_channels)]) + self.downsamplers = nn.ModuleList([Downsample2D(out_channels)]) else: self.downsamplers = None