|
46 | 46 | from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline |
47 | 47 | from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline |
48 | 48 | from .pixart_alpha import PixArtAlphaPipeline |
| 49 | +from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline |
49 | 50 | from .stable_diffusion import ( |
50 | 51 | StableDiffusionImg2ImgPipeline, |
51 | 52 | StableDiffusionInpaintPipeline, |
|
70 | 71 | ("stable-diffusion-controlnet", StableDiffusionControlNetPipeline), |
71 | 72 | ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline), |
72 | 73 | ("wuerstchen", WuerstchenCombinedPipeline), |
| 74 | + ("cascade", StableCascadeCombinedPipeline), |
73 | 75 | ("lcm", LatentConsistencyModelPipeline), |
74 | 76 | ("pixart", PixArtAlphaPipeline), |
75 | 77 | ] |
|
106 | 108 | ("kandinsky", KandinskyPipeline), |
107 | 109 | ("kandinsky22", KandinskyV22Pipeline), |
108 | 110 | ("wuerstchen", WuerstchenDecoderPipeline), |
| 111 | + ("cascade", StableCascadeDecoderPipeline), |
109 | 112 | ] |
110 | 113 | ) |
111 | 114 | _AUTO_IMAGE2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict( |
|
0 commit comments