Skip to content

GitLab

Unverified Commit 24bdf4b2 authored by Beinsezii's avatar Beinsezii Committed by GitHub
Browse files

Add SD3 AutoPipeline mappings (#8489)

parent 95e0c375
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 24bdf4b2
...@@ -52,6 +52,10 @@ from .stable_diffusion import ( ...@@ -52,6 +52,10 @@ from .stable_diffusion import (
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipeline,
StableDiffusionPipeline, StableDiffusionPipeline,
) )
from .stable_diffusion_3 import (
StableDiffusion3Img2ImgPipeline,
StableDiffusion3Pipeline,
)
from .stable_diffusion_xl import ( from .stable_diffusion_xl import (
StableDiffusionXLImg2ImgPipeline, StableDiffusionXLImg2ImgPipeline,
StableDiffusionXLInpaintPipeline, StableDiffusionXLInpaintPipeline,
...@@ -64,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -64,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
[ [
("stable-diffusion", StableDiffusionPipeline), ("stable-diffusion", StableDiffusionPipeline),
("stable-diffusion-xl", StableDiffusionXLPipeline), ("stable-diffusion-xl", StableDiffusionXLPipeline),
("stable-diffusion-3", StableDiffusion3Pipeline),
("if", IFPipeline), ("if", IFPipeline),
("kandinsky", KandinskyCombinedPipeline), ("kandinsky", KandinskyCombinedPipeline),
("kandinsky22", KandinskyV22CombinedPipeline), ("kandinsky22", KandinskyV22CombinedPipeline),
...@@ -82,6 +87,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -82,6 +87,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict(
[ [
("stable-diffusion", StableDiffusionImg2ImgPipeline), ("stable-diffusion", StableDiffusionImg2ImgPipeline),
("stable-diffusion-xl", StableDiffusionXLImg2ImgPipeline), ("stable-diffusion-xl", StableDiffusionXLImg2ImgPipeline),
("stable-diffusion-3", StableDiffusion3Img2ImgPipeline),
("if", IFImg2ImgPipeline), ("if", IFImg2ImgPipeline),
("kandinsky", KandinskyImg2ImgCombinedPipeline), ("kandinsky", KandinskyImg2ImgCombinedPipeline),
("kandinsky22", KandinskyV22Img2ImgCombinedPipeline), ("kandinsky22", KandinskyV22Img2ImgCombinedPipeline),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment