Skip to content

Commit 24bdf4b

Browse files
authored
Add SD3 AutoPipeline mappings (#8489)
1 parent 95e0c37 commit 24bdf4b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@
5252
StableDiffusionInpaintPipeline,
5353
StableDiffusionPipeline,
5454
)
55+
from .stable_diffusion_3 import (
56+
StableDiffusion3Img2ImgPipeline,
57+
StableDiffusion3Pipeline,
58+
)
5559
from .stable_diffusion_xl import (
5660
StableDiffusionXLImg2ImgPipeline,
5761
StableDiffusionXLInpaintPipeline,
@@ -64,6 +68,7 @@
6468
[
6569
("stable-diffusion", StableDiffusionPipeline),
6670
("stable-diffusion-xl", StableDiffusionXLPipeline),
71+
("stable-diffusion-3", StableDiffusion3Pipeline),
6772
("if", IFPipeline),
6873
("kandinsky", KandinskyCombinedPipeline),
6974
("kandinsky22", KandinskyV22CombinedPipeline),
@@ -82,6 +87,7 @@
8287
[
8388
("stable-diffusion", StableDiffusionImg2ImgPipeline),
8489
("stable-diffusion-xl", StableDiffusionXLImg2ImgPipeline),
90+
("stable-diffusion-3", StableDiffusion3Img2ImgPipeline),
8591
("if", IFImg2ImgPipeline),
8692
("kandinsky", KandinskyImg2ImgCombinedPipeline),
8793
("kandinsky22", KandinskyV22Img2ImgCombinedPipeline),

0 commit comments

Comments
 (0)