Add AuraFlowPipeline and KolorsPipeline to auto

Skquark · Jul 14, 2024 · 56aaa7b · 56aaa7b
1 parent df0bf24
commit 56aaa7b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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),
     ]
 )