From 4e36fd293966d28f9a6d9dc9775bd5e4fbec9b71 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Wed, 5 Jul 2023 23:06:04 +0900 Subject: [PATCH] fix: ad_ images save file name --- scripts/!adetailer.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 53d8666..d0adf22 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -414,7 +414,12 @@ class AfterDetailerScript(scripts.Script): return i2i def save_image(self, p, image, *, condition: str, suffix: str) -> None: - i = p._ad_idx + i = p._ad_idx_all + if p.all_prompts: + i %= len(p.all_prompts) + save_prompt = p.all_prompts[i] + else: + save_prompt = p.prompt seed, _ = self.get_seed(p) if opts.data.get(condition, False): @@ -423,7 +428,7 @@ class AfterDetailerScript(scripts.Script): path=p.outpath_samples, basename="", seed=seed, - prompt=p.all_prompts[i] if i < len(p.all_prompts) else p.prompt, + prompt=save_prompt, extension=opts.samples_format, info=self.infotext(p), p=p,