diff options
| author | Greg Fuller <gfuller23@gmail.com> | 2022-10-12 23:48:53 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-13 10:25:29 +0000 |
| commit | a3f02e4690844715a510b7bc857a0971dd05c4d8 (patch) | |
| tree | f2d6b9f6a67f8947ceac6746b863330286d2e3f9 | |
| parent | 8711c2fe0135d5c160a57db41cb79ed1942ce7fa (diff) | |
| download | stable-diffusion-webui-gfx803-a3f02e4690844715a510b7bc857a0971dd05c4d8.tar.gz stable-diffusion-webui-gfx803-a3f02e4690844715a510b7bc857a0971dd05c4d8.tar.bz2 stable-diffusion-webui-gfx803-a3f02e4690844715a510b7bc857a0971dd05c4d8.zip | |
fix prompt in log.csv
| -rw-r--r-- | modules/ui.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/modules/ui.py b/modules/ui.py index e07ee0e1..edb4dab1 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -139,6 +139,8 @@ def save_files(js_data, images, do_make_zip, index): if at_start:
writer.writerow(["prompt", "seed", "width", "height", "sampler", "cfgs", "steps", "filename", "negative_prompt"])
+ log_prompt=data["prompt"]
+ log_seed=data["seed"]
for image_index, filedata in enumerate(images, start_index):
if filedata.startswith("data:image/png;base64,"):
filedata = filedata[len("data:image/png;base64,"):]
@@ -148,7 +150,9 @@ 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)
+ log_seed=p.all_seeds[i]
+ log_prompt=p.all_prompts[i]
+ fullfn, txt_fullfn = save_image(image, path, "", seed=log_seed, prompt=log_prompt, 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)
@@ -157,7 +161,7 @@ def save_files(js_data, images, do_make_zip, index): 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"]])
+ writer.writerow([log_prompt, log_seed, data["width"], data["height"], data["sampler"], data["cfg_scale"], data["steps"], filenames[0], data["negative_prompt"]])
# Make Zip
if do_make_zip:
|