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