fixed issue when executing without batch mode

fjteam · Feb 26, 2023 · 70d4e34 · 70d4e34
1 parent 6247fc5
commit 70d4e34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
@@ -576,8 +576,8 @@ def process(self, p, is_img2img=False, *args):
 
  is_api = getattr(p, 'control_net_api_access', False)
  is_img2img_batch_tab = not is_api and is_img2img and img2img_tab_tracker.submit_img2img_tab == 'img2img_batch_tab'
- if is_img2img_batch_tab and p.image_control is not None:
- input_image = p.image_control 
+ if is_img2img_batch_tab and hasattr(p, "image_control") and p.image_control is not None:
+ input_image = HWC3(np.asarray(p.image_control)) 
  elif input_image is not None:
  input_image = HWC3(np.asarray(input_image))
  elif image is not None: