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"]