quick fix to make sure sdxl conversion works

d0979f52 · Patrick von Platen · fcb0da7f · d0979f52
Commit d0979f52 authored Jul 11, 2023 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py ...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +2 -1

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