Skip to content

GitLab

Unverified Commit 9bafef34 authored by Beinsezii's avatar Beinsezii Committed by GitHub
Browse files

Add Pixart to AUTO_TEXT2IMAGE_PIPELINES_MAPPING (#5664)

parent 64603389
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 9bafef34
...@@ -43,6 +43,7 @@ from .kandinsky2_2 import ( ...@@ -43,6 +43,7 @@ from .kandinsky2_2 import (
KandinskyV22Pipeline, KandinskyV22Pipeline,
) )
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
from .pixart_alpha import PixArtAlphaPipeline
from .stable_diffusion import ( from .stable_diffusion import (
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipeline,
...@@ -67,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -67,6 +68,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline), ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
("wuerstchen", WuerstchenCombinedPipeline), ("wuerstchen", WuerstchenCombinedPipeline),
("lcm", LatentConsistencyModelPipeline), ("lcm", LatentConsistencyModelPipeline),
("pixart", PixArtAlphaPipeline),
] ]
) )
......
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