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

fix(scripts): fix live preview None error

Browse Source
This commit is contained in:
Dowon
2024-04-14 16:46:59 +09:00
parent d8df3b8111
commit 3c27ab30ce
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/!adetailer.py
View File

@@ -736,15 +736,15 @@ class AfterDetailerScript(scripts.Script):
with change_torch_load(): with change_torch_load():
pred = predictor(ad_model, pp.image, args.ad_confidence, **kwargs) pred = predictor(ad_model, pp.image, args.ad_confidence, **kwargs)
masks = self.pred_preprocessing(p, pred, args) if pred.preview is None:
shared.state.assign_current_image(pred.preview)
if not masks:
print( print(
f"[-] ADetailer: nothing detected on image {i + 1} with {ordinal(n + 1)} settings." f"[-] ADetailer: nothing detected on image {i + 1} with {ordinal(n + 1)} settings."
) )
return False return False
masks = self.pred_preprocessing(p, pred, args)
shared.state.assign_current_image(pred.preview)
self.save_image( self.save_image(
p, p,
pred.preview, pred.preview,