From 2cc23b999611cc68f32ef915ffeed1c483ed78ed Mon Sep 17 00:00:00 2001 From: Bingsu Date: Thu, 27 Apr 2023 21:38:57 +0900 Subject: [PATCH] fix: save preview image with info --- scripts/!adetailer.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index ca0966d..f52227d 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -277,10 +277,13 @@ class AfterDetailerScript(scripts.Script): device = "cpu" return device - def write_params_txt(self, p): - infotext = create_infotext( + def infotext(self, p): + return create_infotext( p, p.all_prompts, p.all_seeds, p.all_subseeds, None, 0, 0 ) + + def write_params_txt(self, p): + infotext = self.infotext(p) params_txt = Path(data_path, "params.txt") params_txt.write_text(infotext, encoding="utf-8") @@ -372,12 +375,13 @@ class AfterDetailerScript(scripts.Script): if opts.data.get("ad_save_previews", False): images.save_image( - pred.preview, - p.outpath_samples, - "", - seed, - p.all_prompts[i], - opts.samples_format, + image=pred.preview, + path=p.outpath_samples, + basename="", + seed=seed, + prompt=p.all_prompts[i], + extension=opts.samples_format, + info=self.infotext(p), p=p, suffix="-ad-preview", )