Skip to content

Commit

Permalink
fix issues with controlnet save images
Browse files Browse the repository at this point in the history
  • Loading branch information
xieyongliang committed Mar 27, 2023
1 parent ace896b commit 725dd7a
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions modules/ui.py
Expand Up @@ -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)
Expand Down

0 comments on commit 725dd7a

Please sign in to comment.