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(
checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs
)
pipe = StableDiffusionXLPipeline(
pipe = pipeline_class(
vae=vae,
text_encoder=text_encoder,
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