"docs/vscode:/vscode.git/clone" did not exist on "6246c70d2150dcc58415facfe5c199f49b4d2af1"
Unverified Commit c77ac246 authored by Byron Mallett's avatar Byron Mallett Committed by GitHub
Browse files

Fixed SDXL single file loading to use the correct requested pipeline class (#4142)

Using pipeline_class argument to instantiate the correct pipeline when loading SDXL models from single files
parent ed2a3584
...@@ -1536,7 +1536,7 @@ def download_from_original_stable_diffusion_ckpt( ...@@ -1536,7 +1536,7 @@ def download_from_original_stable_diffusion_ckpt(
checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs
) )
pipe = StableDiffusionXLPipeline( pipe = pipeline_class(
vae=vae, vae=vae,
text_encoder=text_encoder, text_encoder=text_encoder,
tokenizer=tokenizer, tokenizer=tokenizer,
......
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