Skip to content

Commit 5e80827

Browse files
quick fix to make sure sdxl conversion works
1 parent 6894056 commit 5e80827

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1256,7 +1256,8 @@ def download_from_original_stable_diffusion_ckpt(
12561256
elif num_in_channels is None:
12571257
num_in_channels = 4
12581258

1259-
original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels
1259+
if "unet_config" in original_config.model.params:
1260+
original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels
12601261

12611262
if (
12621263
"parameterization" in original_config["model"]["params"]

0 commit comments

Comments
 (0)