quick controlnet fix

Signed-off-by: Vladimir Mandic <[email protected]>
vladmandic · Feb 15, 2025 · 227a419 · 227a419
1 parent 23134eb
commit 227a419
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/modules/control/run.py b/modules/control/run.py
@@ -627,7 +627,7 @@ def control_run(state: str = '',
                             if 'strength' in possible:
                                 p.task_args['strength'] = p.denoising_strength
                             p.init_images = [init_image] * len(active_model)
-                        if hasattr(shared.sd_model, 'controlnet') and len(p.task_args['control_image']) > 1 and (shared.sd_model.__class__.__name__ == 'StableDiffusionXLControlNetUnionPipeline'): # special case for controlnet-union
+                        if hasattr(shared.sd_model, 'controlnet') and hasattr(p.task_args, 'control_image') and len(p.task_args['control_image']) > 1 and (shared.sd_model.__class__.__name__ == 'StableDiffusionXLControlNetUnionPipeline'): # special case for controlnet-union
                             p.task_args['control_image'] = [[x] for x in p.task_args['control_image']]
                             p.task_args['control_mode'] = [[x] for x in p.task_args['control_mode']]
 

diff --git a/wiki b/wiki