From c77ac246c1650f44f0b39b663c14207e7e669bdd Mon Sep 17 00:00:00 2001 From: Byron Mallett Date: Thu, 20 Jul 2023 00:40:13 +1200 Subject: [PATCH] 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 --- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index d37a19b8ad..1c4531dfa7 100644 --- 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,