Add workaround for AutoPipeline.from_pipe with ControlNet (#720)

carson-katri · Nov 1, 2023 · 041302a · 041302a
1 parent 9927174
commit 041302a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/generator_process/actions/load_model.py b/generator_process/actions/load_model.py
@@ -137,8 +137,8 @@ def _convert_pipe(cache, model, pipe, model_class, half_precision, scheduler, **
     if model_class.__name__ not in {
         # some tasks are not supported by auto pipeline
         'DreamTexturesDepth2ImgPipeline',
-        'StableDiffusionUpscalePipeline'
-    }:
+        'StableDiffusionUpscalePipeline',
+    } and not 'controlnet' in kwargs: 
         pipe = model_class.from_pipe(pipe, **kwargs)
     scheduler.create(pipe)
     return pipe