Skip to content

Commit ad0308b

Browse files
Beinseziisayakpaul
andauthored
Add Cascade to Auto T2I + Decoder mappings (huggingface#7362)
* Add Cascade to Auto T2I + Decoder mappings * ruff autofix --------- Co-authored-by: Sayak Paul <[email protected]>
1 parent e97a633 commit ad0308b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
4747
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
4848
from .pixart_alpha import PixArtAlphaPipeline
49+
from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
4950
from .stable_diffusion import (
5051
StableDiffusionImg2ImgPipeline,
5152
StableDiffusionInpaintPipeline,
@@ -70,6 +71,7 @@
7071
("stable-diffusion-controlnet", StableDiffusionControlNetPipeline),
7172
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
7273
("wuerstchen", WuerstchenCombinedPipeline),
74+
("cascade", StableCascadeCombinedPipeline),
7375
("lcm", LatentConsistencyModelPipeline),
7476
("pixart", PixArtAlphaPipeline),
7577
]
@@ -106,6 +108,7 @@
106108
("kandinsky", KandinskyPipeline),
107109
("kandinsky22", KandinskyV22Pipeline),
108110
("wuerstchen", WuerstchenDecoderPipeline),
111+
("cascade", StableCascadeDecoderPipeline),
109112
]
110113
)
111114
_AUTO_IMAGE2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict(

0 commit comments

Comments
 (0)