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(script): missing get_ad_model

Browse Source
This commit is contained in:
Dowon
2024-08-25 15:13:25 +09:00
parent 67793f076c
commit fabb7ded2b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/!adetailer.py
View File

@@ -818,8 +818,9 @@ class AfterDetailerScript(scripts.Script):
else: else:
with change_torch_load(): with change_torch_load():
ad_model = self.get_ad_model(args.ad_model)
pred = ultralytics_predict( pred = ultralytics_predict(
args.ad_model, ad_model,
image=pp.image, image=pp.image,
confidence=args.ad_confidence, confidence=args.ad_confidence,
device=self.ultralytics_device, device=self.ultralytics_device,