Skip to content

Commit 7f74c09

Browse files
mazharosamaDN6
authored andcommitted
Enable CivitAI SDXL Inpainting Models Conversion (#8795)
modify in_channels in network_config params Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
1 parent d1e1676 commit 7f74c09

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1370,6 +1370,8 @@ def download_from_original_stable_diffusion_ckpt(
13701370

13711371
if "unet_config" in original_config["model"]["params"]:
13721372
original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels
1373+
elif "network_config" in original_config["model"]["params"]:
1374+
original_config["model"]["params"]["network_config"]["params"]["in_channels"] = num_in_channels
13731375

13741376
if (
13751377
"parameterization" in original_config["model"]["params"]

0 commit comments

Comments
 (0)