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: batch index

Browse Source
This commit is contained in:
Bingsu
2023-07-18 20:19:00 +09:00
parent b2b64fe288
commit 88de22b4eb
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/!adetailer.py
View File

@@ -498,14 +498,15 @@ class AfterDetailerScript(scripts.Script):
@staticmethod
def need_call_process(p) -> bool:
i = p.batch_index
i = p._ad_idx
bs = p.batch_size
return i == bs - 1
return i % bs == bs - 1
@staticmethod
def need_call_postprocess(p) -> bool:
i = p.batch_index
return i == 0
i = p._ad_idx
bs = p.batch_size
return i % bs == 0
@rich_traceback
def process(self, p, *args_):