experimental segmoe support

This commit is contained in:
Vladimir Mandic
2024-02-05 10:38:42 -05:00
parent ff2c1db1cc
commit e32220ccc1
13 changed files with 1376 additions and 15 deletions
+2 -1
View File
@@ -54,7 +54,8 @@ def get_pipelines():
'ONNX Stable Diffusion XL': getattr(diffusers, 'OnnxStableDiffusionXLPipeline', None),
'ONNX Stable Diffusion XL Img2Img': getattr(diffusers, 'OnnxStableDiffusionXLImg2ImgPipeline', None),
'Custom Diffusers Pipeline': getattr(diffusers, 'DiffusionPipeline', None),
'InstaFlow': getattr(diffusers, 'StableDiffusionPipeline', None) # dynamically redefined and loaded in sd_models.load_diffuser
'InstaFlow': getattr(diffusers, 'StableDiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'SegMoE': getattr(diffusers, 'StableDiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
# Segmind SSD-1B, Segmind Tiny
}