From 725dd7af4f657bddca2d2bd87f6d3aa82ff37979 Mon Sep 17 00:00:00 2001
From: xieyongliang <yongliangxie@hotmail.com>
Date: Tue, 28 Mar 2023 06:44:59 +0800
Subject: [PATCH] fix issues with controlnet save images

---
 modules/ui.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/ui.py b/modules/ui.py
index ccdc322d1ce..1bbc11b0e97 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -157,6 +157,9 @@ def __init__(self, d=None):
             is_grid = image_index < p.index_of_first_image
             i = 0 if is_grid else (image_index - p.index_of_first_image)
 
+            if image_index >= len(p.infotexts) or i >= len(p.all_seeds) or i >= len(p.all_prompts):
+                break
+
             fullfn, txt_fullfn = save_image(image, path, "", seed=p.all_seeds[i], prompt=p.all_prompts[i], extension=extension, info=p.infotexts[image_index], grid=is_grid, p=p, save_to_dirs=save_to_dirs)
 
             filename = os.path.relpath(fullfn, path)