diff --git a/modules/shared_items.py b/modules/shared_items.py index 11e7659d1..82deed604 100644 --- a/modules/shared_items.py +++ b/modules/shared_items.py @@ -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__}')