Skip to content

Commit 4319e83

Browse files
committed
add Sana to auto-text2image-pipeline;
1 parent 5cda8ea commit 4319e83

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
from .pag import (
6969
HunyuanDiTPAGPipeline,
7070
PixArtSigmaPAGPipeline,
71+
SanaPAGPipeline,
7172
StableDiffusion3PAGImg2ImgPipeline,
7273
StableDiffusion3PAGPipeline,
7374
StableDiffusionControlNetPAGInpaintPipeline,
@@ -82,6 +83,7 @@
8283
StableDiffusionXLPAGPipeline,
8384
)
8485
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
86+
from .sana import SanaPipeline
8587
from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
8688
from .stable_diffusion import (
8789
StableDiffusionImg2ImgPipeline,
@@ -121,6 +123,8 @@
121123
("lcm", LatentConsistencyModelPipeline),
122124
("pixart-alpha", PixArtAlphaPipeline),
123125
("pixart-sigma", PixArtSigmaPipeline),
126+
("sana", SanaPipeline),
127+
("sana-pag", SanaPAGPipeline),
124128
("stable-diffusion-pag", StableDiffusionPAGPipeline),
125129
("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGPipeline),
126130
("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),

0 commit comments

Comments
 (0)