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

Mystfit · web-flow · commit c77ac246c165 · 2023-07-19T14:40:13.000+02:00
Using pipeline_class argument to instantiate the correct pipeline when loading SDXL models from single files
diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
@@ -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,

Original file line number	Diff line number	Diff line change
`@@ -1536,7 +1536,7 @@ def download_from_original_stable_diffusion_ckpt(`
`1536`	`1536`	`checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs`
`1537`	`1537`	`)`
`1538`	`1538`
`1539`		`- pipe = StableDiffusionXLPipeline(`
	`1539`	`+ pipe = pipeline_class(`
`1540`	`1540`	`vae=vae,`
`1541`	`1541`	`text_encoder=text_encoder,`
`1542`	`1542`	`tokenizer=tokenizer,`