From 94b332c47665cb85b22a54545d7fb7f6478ea72d Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Tue, 1 Aug 2023 07:47:22 -1000 Subject: [PATCH] support from_single_file for SDXL inpainting (#4408) fix Co-authored-by: yiyixuxu --- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index 07604d7c08..fdbe1dfaef 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1186,7 +1186,6 @@ def download_from_original_stable_diffusion_ckpt( StableDiffusionInpaintPipeline, StableDiffusionPipeline, StableDiffusionXLImg2ImgPipeline, - StableDiffusionXLPipeline, StableUnCLIPImg2ImgPipeline, StableUnCLIPPipeline, ) @@ -1543,7 +1542,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,