Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-02-27 10:34:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: _postprocess_image

Browse Source
This commit is contained in:
Bingsu
2023-05-05 13:54:33 +09:00
parent 9250422ddf
commit 83ebb5c294
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
scripts/!adetailer.py
View File

@@ -448,15 +448,7 @@ class AfterDetailerScript(scripts.Script):
extra_params = self.extra_params(args)
p.extra_generation_params.update(extra_params)
def postprocess_image(self, p, pp, *args_):
if getattr(p, "_disable_adetailer", False):
return
args = self.get_args(*args_)
if not self.is_ad_enabled(args):
return
def _postprocess_image(self, p, pp, args: ADetailerArgs):
p._idx = getattr(p, "_idx", -1) + 1
i = p._idx
@@ -517,8 +509,19 @@ class AfterDetailerScript(scripts.Script):
if processed is not None:
pp.image = processed.images[0]
def postprocess_image(self, p, pp, *args_):
if getattr(p, "_disable_adetailer", False):
return
args = self.get_args(*args_)
if not self.is_ad_enabled(args):
return
self._postprocess_image(p, pp, args)
try:
if i == len(p.all_prompts) - 1:
if p._idx == len(p.all_prompts) - 1:
self.write_params_txt(p)
except Exception:
pass