remove dependency on diffusers dev

This commit is contained in:
Vladimir Mandic
2024-01-31 10:27:15 -05:00
parent 09d1f1e788
commit 778f54e7c7
+5 -1
View File
@@ -47,9 +47,13 @@ def get_pipelines():
'Kandinsky 3': getattr(diffusers, 'Kandinsky3Pipeline', None),
'DeepFloyd IF': getattr(diffusers, 'IFPipeline', None),
'Custom Diffusers Pipeline': getattr(diffusers, 'DiffusionPipeline', None),
'InstaFlow': diffusers.utils.get_class_from_dynamic_module('instaflow_one_step', module_file='pipeline.py')
# Segmind SSD-1B, Segmind Tiny
}
try:
pipelines['instaflow'] = diffusers.utils.get_class_from_dynamic_module('instaflow_one_step', module_file='pipeline.py')
except Exception:
pass
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__}')