diff options
| author | aoirusann <yimodo.siki@gmail.com> | 2022-10-09 13:01:10 +0800 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-09 10:49:11 +0300 |
| commit | 14192c5b207b16b1ec7a4c9c4ea538d1a6811a4d (patch) | |
| tree | ecff2cd17c76c3b52443f788e677f822c836bece /modules/ui.py | |
| parent | 5ab7e88d9b0bb0125af9f7237242a00a93360ce5 (diff) | |
Support `Download` for txt files.
Diffstat (limited to 'modules/ui.py')
| -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 fe7f10a7..debd8873 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -138,11 +138,14 @@ 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 = 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)
+ 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)
filenames.append(filename)
fullfns.append(fullfn)
+ if txt_fullfn:
+ filenames.append(os.path.basename(txt_fullfn))
+ fullfns.append(txt_fullfn)
writer.writerow([data["prompt"], data["seed"], data["width"], data["height"], data["sampler"], data["cfg_scale"], data["steps"], filenames[0], data["negative_prompt"]])
|