post merge cleanup

This commit is contained in:
Vladimir Mandic
2024-01-31 16:02:29 -05:00
parent 016fd03a38
commit d6885a59eb
5 changed files with 20 additions and 11 deletions
+1 -4
View File
@@ -56,12 +56,9 @@ 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
# Segmind SSD-1B, Segmind Tiny
}
try:
pipelines['InstaFlow'] = diffusers.utils.get_class_from_dynamic_module('instaflow_one_step', module_file='pipeline.py')
except Exception:
pipelines['InstaFlow'] = getattr(diffusers, 'StableDiffusionPipeline', None)
for k, v in pipelines.items():
if k != 'Autodetect' and v is None: