diff --git a/modules/shared_items.py b/modules/shared_items.py index 82deed604..4438f4647 100644 --- a/modules/shared_items.py +++ b/modules/shared_items.py @@ -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():