Skip to content

GitLab

Commit d0979f52 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

quick fix to make sure sdxl conversion works

parent fcb0da7f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View file @ d0979f52
...@@ -1256,6 +1256,7 @@ def download_from_original_stable_diffusion_ckpt( ...@@ -1256,6 +1256,7 @@ def download_from_original_stable_diffusion_ckpt(
elif num_in_channels is None: elif num_in_channels is None:
num_in_channels = 4 num_in_channels = 4
if "unet_config" in original_config.model.params:
original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels original_config["model"]["params"]["unet_config"]["params"]["in_channels"] = num_in_channels
if ( if (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment