quick fix to make sure sdxl conversion works

huggingface · Jul 11, 2023 · 5e80827 · 5e80827
1 parent 6894056
commit 5e80827
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"]