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

Add Cascade to Auto T2I + Decoder mappings (#7362)



* Add Cascade to Auto T2I + Decoder mappings

* ruff autofix

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent e97a633b
...@@ -46,6 +46,7 @@ from .kandinsky2_2 import ( ...@@ -46,6 +46,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
from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
from .stable_diffusion import ( from .stable_diffusion import (
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipeline,
...@@ -70,6 +71,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -70,6 +71,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-controlnet", StableDiffusionControlNetPipeline), ("stable-diffusion-controlnet", StableDiffusionControlNetPipeline),
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline), ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
("wuerstchen", WuerstchenCombinedPipeline), ("wuerstchen", WuerstchenCombinedPipeline),
("cascade", StableCascadeCombinedPipeline),
("lcm", LatentConsistencyModelPipeline), ("lcm", LatentConsistencyModelPipeline),
("pixart", PixArtAlphaPipeline), ("pixart", PixArtAlphaPipeline),
] ]
...@@ -106,6 +108,7 @@ _AUTO_TEXT2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict( ...@@ -106,6 +108,7 @@ _AUTO_TEXT2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict(
("kandinsky", KandinskyPipeline), ("kandinsky", KandinskyPipeline),
("kandinsky22", KandinskyV22Pipeline), ("kandinsky22", KandinskyV22Pipeline),
("wuerstchen", WuerstchenDecoderPipeline), ("wuerstchen", WuerstchenDecoderPipeline),
("cascade", StableCascadeDecoderPipeline),
] ]
) )
_AUTO_IMAGE2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict( _AUTO_IMAGE2IMAGE_DECODER_PIPELINES_MAPPING = OrderedDict(
......
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