From fabb7ded2b7297c63b4138f9c521efa2ad0d1f5a Mon Sep 17 00:00:00 2001 From: Dowon Date: Sun, 25 Aug 2024 15:13:25 +0900 Subject: [PATCH] fix(script): missing get_ad_model --- scripts/!adetailer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 908c1ea..b4156fc 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -818,8 +818,9 @@ def _postprocess_image_inner( else: with change_torch_load(): + ad_model = self.get_ad_model(args.ad_model) pred = ultralytics_predict( - args.ad_model, + ad_model, image=pp.image, confidence=args.ad_confidence, device=self.ultralytics_device,