fix(script): missing get_ad_model

Bing-su · Aug 25, 2024 · fabb7de · fabb7de
1 parent 67793f0
commit fabb7de
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,