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

fix: wrong index issue

Browse Source
This commit is contained in:
Bingsu
2023-07-07 10:00:21 +09:00
parent 5e4ac4af8c
commit e61b806757
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/!adetailer.py
View File

@@ -414,7 +414,7 @@ class AfterDetailerScript(scripts.Script):
return i2i
def save_image(self, p, image, *, condition: str, suffix: str) -> None:
i = p._ad_idx_all
i = p._ad_idx
if p.all_prompts:
i %= len(p.all_prompts)
save_prompt = p.all_prompts[i]
@@ -482,12 +482,12 @@ class AfterDetailerScript(scripts.Script):
)
def need_call_process(self, p) -> bool:
i = p._ad_idx
i = p._ad_idx_batch
bs = p.batch_size
return i == bs - 1
def need_call_postprocess(self, p) -> bool:
i = p._ad_idx
i = p._ad_idx_batch
return i == 0
@rich_traceback
@@ -500,7 +500,7 @@ class AfterDetailerScript(scripts.Script):
extra_params = self.extra_params(arg_list)
p.extra_generation_params.update(extra_params)
p._ad_idx = -1
p._ad_idx_batch = -1
def _postprocess_image(self, p, pp, args: ADetailerArgs, *, n: int = 0) -> bool:
"""
@@ -597,7 +597,7 @@ class AfterDetailerScript(scripts.Script):
return
p._ad_idx = getattr(p, "_ad_idx", -1) + 1
p._ad_idx_all = getattr(p, "_ad_idx_all", -1) + 1
p._ad_idx_batch = getattr(p, "_ad_idx_batch", -1) + 1
init_image = copy(pp.image)
arg_list = self.get_args(p, *args_)
@@ -623,7 +623,7 @@ class AfterDetailerScript(scripts.Script):
p.scripts.process(copy(p))
try:
ia = p._ad_idx_all
ia = p._ad_idx
lenp = len(p.all_prompts)
if ia % lenp == lenp - 1:
self.write_params_txt(p)