From 56aaa7b74ac93a707e0d7a4fa2c33f58147c1c1d Mon Sep 17 00:00:00 2001
From: Alan Bedian <skquark@yahoo.com>
Date: Sun, 14 Jul 2024 00:50:48 -0700
Subject: [PATCH] Add AuraFlowPipeline and KolorsPipeline to auto

---
 src/diffusers/pipelines/auto_pipeline.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py
index 1369d3e80c57..7fc8a8e78028 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, KolorsImg2ImgPipeline
 from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
 from .pag import (
     StableDiffusionControlNetPAGPipeline,
@@ -95,6 +97,8 @@
         ("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
         ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
         ("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline),
+        ("auraflow", AuraFlowPipeline),
+        ("kolors", KolorsPipeline),
     ]
 )
 
@@ -111,6 +115,7 @@
         ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline),
         ("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline),
         ("lcm", LatentConsistencyModelImg2ImgPipeline),
+        ("kolors", KolorsImg2ImgPipeline),
     ]
 )