Skip to content

Commit 28cf8be

Browse files
Beinseziiyiyixuxu
authored andcommitted
Add AuraFlowPipeline and KolorsPipeline to auto map (huggingface#8849)
* Add AuraFlowPipeline and KolorsPipeline to auto map Just T2I. Validated using `quickdif` * Add Kolors I2I and SD3 Inpaint auto maps * style --------- Co-authored-by: yiyixuxu <yixu310@gmail.com>
1 parent 15657ce commit 28cf8be

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from huggingface_hub.utils import validate_hf_hub_args
1919

2020
from ..configuration_utils import ConfigMixin
21+
from .aura_flow import AuraFlowPipeline
2122
from .controlnet import (
2223
StableDiffusionControlNetImg2ImgPipeline,
2324
StableDiffusionControlNetInpaintPipeline,
@@ -45,6 +46,7 @@
4546
KandinskyV22Pipeline,
4647
)
4748
from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
49+
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
4850
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
4951
from .pag import (
5052
StableDiffusionControlNetPAGPipeline,
@@ -63,6 +65,7 @@
6365
)
6466
from .stable_diffusion_3 import (
6567
StableDiffusion3Img2ImgPipeline,
68+
StableDiffusion3InpaintPipeline,
6669
StableDiffusion3Pipeline,
6770
)
6871
from .stable_diffusion_xl import (
@@ -94,6 +97,8 @@
9497
("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
9598
("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
9699
("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline),
100+
("auraflow", AuraFlowPipeline),
101+
("kolors", KolorsPipeline),
97102
]
98103
)
99104

@@ -110,13 +115,15 @@
110115
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline),
111116
("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline),
112117
("lcm", LatentConsistencyModelImg2ImgPipeline),
118+
("kolors", KolorsImg2ImgPipeline),
113119
]
114120
)
115121

116122
AUTO_INPAINT_PIPELINES_MAPPING = OrderedDict(
117123
[
118124
("stable-diffusion", StableDiffusionInpaintPipeline),
119125
("stable-diffusion-xl", StableDiffusionXLInpaintPipeline),
126+
("stable-diffusion-3", StableDiffusion3InpaintPipeline),
120127
("if", IFInpaintingPipeline),
121128
("kandinsky", KandinskyInpaintCombinedPipeline),
122129
("kandinsky22", KandinskyV22InpaintCombinedPipeline),

0 commit comments

Comments
 (0)