|
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