This commit is contained in:
Vladimir Mandic
2024-01-31 10:29:01 -05:00
parent 778f54e7c7
commit f7842d959e
+2 -1
View File
@@ -50,8 +50,9 @@ def get_pipelines():
# Segmind SSD-1B, Segmind Tiny
}
try:
pipelines['instaflow'] = diffusers.utils.get_class_from_dynamic_module('instaflow_one_step', module_file='pipeline.py')
pipelines['InstaFlow'] = diffusers.utils.get_class_from_dynamic_module('instaflow_one_step', module_file='pipeline.py')
except Exception:
pipelines['InstaFlow'] = getattr(diffusers, 'StableDiffusionPipeline', None)
pass
for k, v in pipelines.items():