Skip to content

Commit 49c9517

Browse files
duongna21patrickvonplaten
authored andcommitted
Fix SDXL conversion from original to diffusers (#4280)
* fix sdxl conversion * convention
1 parent c2f755b commit 49c9517

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1186,6 +1186,7 @@ def download_from_original_stable_diffusion_ckpt(
11861186
StableDiffusionInpaintPipeline,
11871187
StableDiffusionPipeline,
11881188
StableDiffusionXLImg2ImgPipeline,
1189+
StableDiffusionXLPipeline,
11891190
StableUnCLIPImg2ImgPipeline,
11901191
StableUnCLIPPipeline,
11911192
)
@@ -1542,7 +1543,7 @@ def download_from_original_stable_diffusion_ckpt(
15421543
checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs
15431544
)
15441545

1545-
pipe = pipeline_class(
1546+
pipe = StableDiffusionXLPipeline(
15461547
vae=vae,
15471548
text_encoder=text_encoder,
15481549
tokenizer=tokenizer,

0 commit comments

Comments
 (0)