From 778f54e7c72cd3ccec7975f00bb9f64cc54fd281 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 31 Jan 2024 10:27:15 -0500 Subject: [PATCH] remove dependency on diffusers dev --- modules/shared_items.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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__}')