From a298173cf0b5a7af64253a76bcab6123c693f09e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 25 Jul 2025 13:13:32 -0400 Subject: [PATCH] handle batch when saving images Signed-off-by: Vladimir Mandic --- modules/prompt_parser_diffusers.py | 2 +- modules/ui_common.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index 12537fb1c..e72099b98 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -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 diff --git a/modules/ui_common.py b/modules/ui_common.py index 314d9f031..454775e1b 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -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