mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
handle batch when saving images
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user