From 20a7279df751c62051cf657b50a9b23029c055ef Mon Sep 17 00:00:00 2001 From: Beinsezii Date: Thu, 11 Jul 2024 20:30:10 -0700 Subject: [PATCH 1/3] Add AuraFlowPipeline and KolorsPipeline to auto map Just T2I. Validated using `quickdif` --- src/diffusers/pipelines/auto_pipeline.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py index 3eb98cfef912..9d0b4831f440 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -17,6 +17,7 @@ from huggingface_hub.utils import validate_hf_hub_args +from .aura_flow import AuraFlowPipeline from ..configuration_utils import ConfigMixin from .controlnet import ( StableDiffusionControlNetImg2ImgPipeline, @@ -45,6 +46,7 @@ KandinskyV22Pipeline, ) from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline +from .kolors import KolorsPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .pag import ( StableDiffusionPAGPipeline, @@ -92,6 +94,8 @@ ("stable-diffusion-pag", StableDiffusionPAGPipeline), ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline), ("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline), + ("auraflow", AuraFlowPipeline), + ("kolors", KolorsPipeline), ] ) From 306598061cac5e977e74d80fa84445bbcbbe462f Mon Sep 17 00:00:00 2001 From: Beinsezii Date: Sun, 14 Jul 2024 16:35:10 -0700 Subject: [PATCH 2/3] Add Kolors I2I and SD3 Inpaint auto maps --- src/diffusers/pipelines/auto_pipeline.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py index 9d0b4831f440..5e3c176f4e44 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -46,7 +46,7 @@ KandinskyV22Pipeline, ) from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline -from .kolors import KolorsPipeline +from .kolors import KolorsPipeline, KolorsImg2ImgPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .pag import ( StableDiffusionPAGPipeline, @@ -65,6 +65,7 @@ from .stable_diffusion_3 import ( StableDiffusion3Img2ImgPipeline, StableDiffusion3Pipeline, + StableDiffusion3InpaintPipeline, ) from .stable_diffusion_xl import ( StableDiffusionXLImg2ImgPipeline, @@ -112,6 +113,7 @@ ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline), ("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline), ("lcm", LatentConsistencyModelImg2ImgPipeline), + ("kolors", KolorsImg2ImgPipeline), ] ) @@ -119,6 +121,7 @@ [ ("stable-diffusion", StableDiffusionInpaintPipeline), ("stable-diffusion-xl", StableDiffusionXLInpaintPipeline), + ("stable-diffusion-3", StableDiffusion3InpaintPipeline), ("if", IFInpaintingPipeline), ("kandinsky", KandinskyInpaintCombinedPipeline), ("kandinsky22", KandinskyV22InpaintCombinedPipeline), From 78e6020a7dccc0b67a177424affd6373674095ae Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Wed, 17 Jul 2024 03:36:25 +0200 Subject: [PATCH 3/3] style --- src/diffusers/pipelines/auto_pipeline.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py index 547e0cf85f2e..be714a657c10 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -17,8 +17,8 @@ from huggingface_hub.utils import validate_hf_hub_args -from .aura_flow import AuraFlowPipeline from ..configuration_utils import ConfigMixin +from .aura_flow import AuraFlowPipeline from .controlnet import ( StableDiffusionControlNetImg2ImgPipeline, StableDiffusionControlNetInpaintPipeline, @@ -46,7 +46,7 @@ KandinskyV22Pipeline, ) from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline -from .kolors import KolorsPipeline, KolorsImg2ImgPipeline +from .kolors import KolorsImg2ImgPipeline, KolorsPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .pag import ( StableDiffusionControlNetPAGPipeline, @@ -65,8 +65,8 @@ ) from .stable_diffusion_3 import ( StableDiffusion3Img2ImgPipeline, - StableDiffusion3Pipeline, StableDiffusion3InpaintPipeline, + StableDiffusion3Pipeline, ) from .stable_diffusion_xl import ( StableDiffusionXLImg2ImgPipeline,