From 3b24fe36a727b125f68cdac32a1ff48f1f2920e1 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Thu, 26 Jan 2023 20:10:58 +0100 Subject: [PATCH 1/2] Fix downsample channels. (#110) 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 @@ class CrossAttnDownBlock2D(nn.Module): 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 From 6cd5c7a76003f7f7326b91b7480fee59398dc7ef Mon Sep 17 00:00:00 2001 From: Stephen <103328915+stephengoodman@users.noreply.github.com> Date: Thu, 26 Jan 2023 14:12:45 -0500 Subject: [PATCH 2/2] Update import to nested path (#109) * Update import to nested path * Update python_coreml_stable_diffusion/unet.py Co-authored-by: Pedro Cuenca Co-authored-by: Atila Orhon <64497909+atiorh@users.noreply.github.com> Co-authored-by: Pedro Cuenca --- 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 f75c7f2..5337503 100644 --- a/python_coreml_stable_diffusion/unet.py +++ b/python_coreml_stable_diffusion/unet.py @@ -6,7 +6,7 @@ from python_coreml_stable_diffusion.layer_norm import LayerNormANE from diffusers.configuration_utils import ConfigMixin, register_to_config -from diffusers.modeling_utils import ModelMixin +from diffusers import ModelMixin from enum import Enum