diff options
| author | Greg Fuller <gfuller23@gmail.com> | 2022-10-11 19:46:33 -0700 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-13 13:25:29 +0300 |
| commit | aeacbac218c47f61f1d0d3f3b429c9038b8faf0f (patch) | |
| tree | e9d136c7881620563cbf756c5fd2f053f4bf1bfc /modules | |
| parent | 94c01aa35656130b56f401830ad443ce3d97c364 (diff) | |
Fix save error
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/ui.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/ui.py b/modules/ui.py index e07ee0e1..4fa405a9 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -148,7 +148,10 @@ def save_files(js_data, images, do_make_zip, index): is_grid = image_index < p.index_of_first_image
i = 0 if is_grid else (image_index - p.index_of_first_image)
- 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)
+ seed = p.all_seeds[i] if len(p.all_seeds) > 1 else p.seed
+ prompt = p.all_prompts[i] if len(p.all_prompts) > 1 else p.prompt
+ info = p.infotexts[image_index] if len(p.infotexts) > 1 else p.infotexts[0]
+ fullfn, txt_fullfn = save_image(image, path, "", seed=seed, prompt=prompt, extension=extension, info=info, grid=is_grid, p=p, save_to_dirs=save_to_dirs)
filename = os.path.relpath(fullfn, path)
filenames.append(filename)
|