Skip to content

GitLab

Unverified Commit ae019da9 authored by Junsong Chen's avatar Junsong Chen Committed by GitHub
Browse files

[Sana] add Sana to auto-text2image-pipeline; (#10538) 

add Sana to auto-text2image-pipeline;
parent 329771e5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ ae019da9
...@@ -68,6 +68,7 @@ from .lumina import LuminaText2ImgPipeline ...@@ -68,6 +68,7 @@ from .lumina import LuminaText2ImgPipeline
from .pag import ( from .pag import (
HunyuanDiTPAGPipeline, HunyuanDiTPAGPipeline,
PixArtSigmaPAGPipeline, PixArtSigmaPAGPipeline,
SanaPAGPipeline,
StableDiffusion3PAGImg2ImgPipeline, StableDiffusion3PAGImg2ImgPipeline,
StableDiffusion3PAGPipeline, StableDiffusion3PAGPipeline,
StableDiffusionControlNetPAGInpaintPipeline, StableDiffusionControlNetPAGInpaintPipeline,
...@@ -82,6 +83,7 @@ from .pag import ( ...@@ -82,6 +83,7 @@ from .pag import (
StableDiffusionXLPAGPipeline, StableDiffusionXLPAGPipeline,
) )
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
from .sana import SanaPipeline
from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
from .stable_diffusion import ( from .stable_diffusion import (
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
...@@ -121,6 +123,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -121,6 +123,8 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("lcm", LatentConsistencyModelPipeline), ("lcm", LatentConsistencyModelPipeline),
("pixart-alpha", PixArtAlphaPipeline), ("pixart-alpha", PixArtAlphaPipeline),
("pixart-sigma", PixArtSigmaPipeline), ("pixart-sigma", PixArtSigmaPipeline),
("sana", SanaPipeline),
("sana-pag", SanaPAGPipeline),
("stable-diffusion-pag", StableDiffusionPAGPipeline), ("stable-diffusion-pag", StableDiffusionPAGPipeline),
("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline), ("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline), ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
......
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