handle batch when saving images

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-07-25 13:13:32 -04:00
parent b311bdf91e
commit a298173cf0
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -253,7 +253,7 @@ class PromptEmbedder:
return torch.cat(res)
except Exception as e:
shared.log.error(f"Prompt encode: {e}")
errors.display(e, 'encode')
# errors.display(e, 'encode')
return None
+2 -2
View File
@@ -174,8 +174,8 @@ def save_files(js_data, files, html_info, index):
items = infotext.parse(geninfo)
p = PObject(items)
try:
seed = p.all_seeds[i]
prompt = p.all_prompts[i]
seed = p.all_seeds[i] if i < len(p.all_seeds) else p.seed
prompt = p.all_prompts[i] if i < len(p.all_prompts) else p.prompt
fullfn, txt_fullfn, _exif = images.save_image(image, shared.opts.outdir_save, "", seed=seed, prompt=prompt, info=info, extension=shared.opts.samples_format, grid=is_grid, p=p)
except Exception as e:
fullfn, txt_fullfn = None, None