fix stable cascade

This commit is contained in:
Vladimir Mandic
2024-05-14 14:20:13 -04:00
parent 1ad27d4b25
commit f7fa2794df
2 changed files with 30 additions and 30 deletions
+2 -6
View File
@@ -62,8 +62,10 @@ def get_pipelines():
'Stable Diffusion XL Img2Img': getattr(diffusers, 'StableDiffusionXLImg2ImgPipeline', None),
'Stable Diffusion XL Inpaint': getattr(diffusers, 'StableDiffusionXLInpaintPipeline', None),
'Stable Diffusion XL Instruct': getattr(diffusers, 'StableDiffusionXLInstructPix2PixPipeline', None),
'Stable Cascade': getattr(diffusers, 'StableCascadeCombinedPipeline', None),
'Latent Consistency Model': getattr(diffusers, 'LatentConsistencyModelPipeline', None),
'PixArt-Alpha': getattr(diffusers, 'PixArtAlphaPipeline', None),
'PixArt-Sigma': getattr(diffusers, 'PixArtSigmaPipeline', None),
'UniDiffuser': getattr(diffusers, 'UniDiffuserPipeline', None),
'Wuerstchen': getattr(diffusers, 'WuerstchenCombinedPipeline', None),
'Kandinsky 2.1': getattr(diffusers, 'KandinskyPipeline', None),
@@ -85,12 +87,6 @@ def get_pipelines():
}
pipelines.update(onnx_pipelines)
if hasattr(diffusers, 'StableCascadeCombinedPipeline'):
pipelines['Stable Cascade'] = getattr(diffusers, 'StableCascadeCombinedPipeline', None)
from modules.sd_hijack_pixart import PixArtSigmaPipeline
pipelines['PixArt-Sigma'] = PixArtSigmaPipeline
for k, v in pipelines.items():
if k != 'Autodetect' and v is None:
log.error(f'Not available: pipeline={k} diffusers={diffusers.__version__} path={diffusers.__file__}')