1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

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
This commit is contained in:
Byron Mallett
2023-07-20 00:40:13 +12:00
committed by GitHub
parent ed2a3584ab
commit c77ac246c1

View File

@@ -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,