Skip to content

GitLab

Unverified Commit 0d2d424f authored by Beinsezii's avatar Beinsezii Committed by GitHub
Browse files

Add PixArtSigmaPipeline to AutoPipeline mapping (#7783)

parent e24e54fd
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 0d2d424f
...@@ -45,7 +45,7 @@ from .kandinsky2_2 import ( ...@@ -45,7 +45,7 @@ from .kandinsky2_2 import (
) )
from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
from .pixart_alpha import PixArtAlphaPipeline from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
from .stable_diffusion import ( from .stable_diffusion import (
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
...@@ -73,7 +73,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -73,7 +73,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("wuerstchen", WuerstchenCombinedPipeline), ("wuerstchen", WuerstchenCombinedPipeline),
("cascade", StableCascadeCombinedPipeline), ("cascade", StableCascadeCombinedPipeline),
("lcm", LatentConsistencyModelPipeline), ("lcm", LatentConsistencyModelPipeline),
("pixart", PixArtAlphaPipeline), ("pixart-alpha", PixArtAlphaPipeline),
("pixart-sigma", PixArtSigmaPipeline),
] ]
) )
......
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