Add Pixart to AUTO_TEXT2IMAGE_PIPELINES_MAPPING (#5664)

9bafef34 · Beinsezii · GitHub · 64603389 · 9bafef34
Unverified Commit 9bafef34 authored Nov 06, 2023 by Beinsezii Committed by GitHub Nov 07, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/diffusers/pipelines/auto_pipeline.py src/diffusers/pipelines/auto_pipeline.py +2 -0

No files found.
--- a/src/diffusers/pipelines/auto_pipeline.py
+++ b/src/diffusers/pipelines/auto_pipeline.py
@@ -43,6 +43,7 @@ from .kandinsky2_2 import (
    KandinskyV22Pipeline,
 )
 from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
+from .pixart_alpha import PixArtAlphaPipeline
 from .stable_diffusion import (
    StableDiffusionImg2ImgPipeline,
    StableDiffusionInpaintPipeline,
@@ -67,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
        ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
        ("wuerstchen", WuerstchenCombinedPipeline),
        ("lcm", LatentConsistencyModelPipeline),
+        ("pixart", PixArtAlphaPipeline),
    ]
 )