fix(scripts): fix live preview None error

This commit is contained in:
Dowon
2024-04-14 16:46:59 +09:00
parent d8df3b8111
commit 3c27ab30ce

View File

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