"references/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "7dc5e5bd60b55eb4e6ea5c1265d6dc7b17d2e917"
Unverified Commit e15a8e7f authored by Beinsezii's avatar Beinsezii Committed by GitHub
Browse files

Add AuraFlowPipeline and KolorsPipeline to auto map (#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: default avataryiyixuxu <yixu310@gmail.com>
parent c2fbf8da
...@@ -18,6 +18,7 @@ from collections import OrderedDict ...@@ -18,6 +18,7 @@ from collections import OrderedDict
from huggingface_hub.utils import validate_hf_hub_args from huggingface_hub.utils import validate_hf_hub_args
from ..configuration_utils import ConfigMixin from ..configuration_utils import ConfigMixin
from .aura_flow import AuraFlowPipeline
from .controlnet import ( from .controlnet import (
StableDiffusionControlNetImg2ImgPipeline, StableDiffusionControlNetImg2ImgPipeline,
StableDiffusionControlNetInpaintPipeline, StableDiffusionControlNetInpaintPipeline,
...@@ -45,6 +46,7 @@ from .kandinsky2_2 import ( ...@@ -45,6 +46,7 @@ from .kandinsky2_2 import (
KandinskyV22Pipeline, KandinskyV22Pipeline,
) )
from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
from .pag import ( from .pag import (
StableDiffusionControlNetPAGPipeline, StableDiffusionControlNetPAGPipeline,
...@@ -63,6 +65,7 @@ from .stable_diffusion import ( ...@@ -63,6 +65,7 @@ from .stable_diffusion import (
) )
from .stable_diffusion_3 import ( from .stable_diffusion_3 import (
StableDiffusion3Img2ImgPipeline, StableDiffusion3Img2ImgPipeline,
StableDiffusion3InpaintPipeline,
StableDiffusion3Pipeline, StableDiffusion3Pipeline,
) )
from .stable_diffusion_xl import ( from .stable_diffusion_xl import (
...@@ -94,6 +97,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -94,6 +97,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline), ("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline), ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline), ("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline),
("auraflow", AuraFlowPipeline),
("kolors", KolorsPipeline),
] ]
) )
...@@ -110,6 +115,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -110,6 +115,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline), ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline),
("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline), ("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline),
("lcm", LatentConsistencyModelImg2ImgPipeline), ("lcm", LatentConsistencyModelImg2ImgPipeline),
("kolors", KolorsImg2ImgPipeline),
] ]
) )
...@@ -117,6 +123,7 @@ AUTO_INPAINT_PIPELINES_MAPPING = OrderedDict( ...@@ -117,6 +123,7 @@ AUTO_INPAINT_PIPELINES_MAPPING = OrderedDict(
[ [
("stable-diffusion", StableDiffusionInpaintPipeline), ("stable-diffusion", StableDiffusionInpaintPipeline),
("stable-diffusion-xl", StableDiffusionXLInpaintPipeline), ("stable-diffusion-xl", StableDiffusionXLInpaintPipeline),
("stable-diffusion-3", StableDiffusion3InpaintPipeline),
("if", IFInpaintingPipeline), ("if", IFInpaintingPipeline),
("kandinsky", KandinskyInpaintCombinedPipeline), ("kandinsky", KandinskyInpaintCombinedPipeline),
("kandinsky22", KandinskyV22InpaintCombinedPipeline), ("kandinsky22", KandinskyV22InpaintCombinedPipeline),
......
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