|
18 | 18 | from huggingface_hub.utils import validate_hf_hub_args
|
19 | 19 |
|
20 | 20 | from ..configuration_utils import ConfigMixin
|
| 21 | +from .aura_flow import AuraFlowPipeline |
21 | 22 | from .controlnet import (
|
22 | 23 | StableDiffusionControlNetImg2ImgPipeline,
|
23 | 24 | StableDiffusionControlNetInpaintPipeline,
|
|
45 | 46 | KandinskyV22Pipeline,
|
46 | 47 | )
|
47 | 48 | from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
|
| 49 | +from .kolors import KolorsImg2ImgPipeline, KolorsPipeline |
48 | 50 | from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
|
49 | 51 | from .pag import (
|
50 | 52 | StableDiffusionControlNetPAGPipeline,
|
|
63 | 65 | )
|
64 | 66 | from .stable_diffusion_3 import (
|
65 | 67 | StableDiffusion3Img2ImgPipeline,
|
| 68 | + StableDiffusion3InpaintPipeline, |
66 | 69 | StableDiffusion3Pipeline,
|
67 | 70 | )
|
68 | 71 | from .stable_diffusion_xl import (
|
|
94 | 97 | ("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
|
95 | 98 | ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
|
96 | 99 | ("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline),
|
| 100 | + ("auraflow", AuraFlowPipeline), |
| 101 | + ("kolors", KolorsPipeline), |
97 | 102 | ]
|
98 | 103 | )
|
99 | 104 |
|
|
110 | 115 | ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline),
|
111 | 116 | ("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline),
|
112 | 117 | ("lcm", LatentConsistencyModelImg2ImgPipeline),
|
| 118 | + ("kolors", KolorsImg2ImgPipeline), |
113 | 119 | ]
|
114 | 120 | )
|
115 | 121 |
|
116 | 122 | AUTO_INPAINT_PIPELINES_MAPPING = OrderedDict(
|
117 | 123 | [
|
118 | 124 | ("stable-diffusion", StableDiffusionInpaintPipeline),
|
119 | 125 | ("stable-diffusion-xl", StableDiffusionXLInpaintPipeline),
|
| 126 | + ("stable-diffusion-3", StableDiffusion3InpaintPipeline), |
120 | 127 | ("if", IFInpaintingPipeline),
|
121 | 128 | ("kandinsky", KandinskyInpaintCombinedPipeline),
|
122 | 129 | ("kandinsky22", KandinskyV22InpaintCombinedPipeline),
|
|
0 commit comments