From d0979f5274e98212f63651aff12872220ea67335 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 11 Jul 2023 16:51:56 +0000 Subject: [PATCH] quick fix to make sure sdxl conversion works --- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index d4b119caaf..a9094cf12f 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1256,7 +1256,8 @@ def download_from_original_stable_diffusion_ckpt( elif num_in_channels is None: num_in_channels = 4 - original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels + if "unet_config" in original_config.model.params: + original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels if ( "parameterization" in original_config["model"]["params"]