Fix SDXL conversion from original to diffusers (#4280)

* fix sdxl conversion * convention

Fix SDXL conversion from original to diffusers (#4280)
* fix sdxl conversion * convention
a2091b70 · Duong A. Nguyen · GitHub · d8bc1a4e · a2091b70
Unverified Commit a2091b70 authored Jul 27, 2023 by Duong A. Nguyen Committed by GitHub Jul 27, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py ...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +2 -1

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
+++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
@@ -1186,6 +1186,7 @@ def download_from_original_stable_diffusion_ckpt(
        StableDiffusionInpaintPipeline,
        StableDiffusionPipeline,
        StableDiffusionXLImg2ImgPipeline,
+        StableDiffusionXLPipeline,
        StableUnCLIPImg2ImgPipeline,
        StableUnCLIPPipeline,
    )
@@ -1542,7 +1543,7 @@ def download_from_original_stable_diffusion_ckpt(
                checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs
            )
-            pipe = pipeline_class(
+            pipe = StableDiffusionXLPipeline(
                vae=vae,
                text_encoder=text_encoder,
                tokenizer=tokenizer,