Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 19:29:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: change unsafe pickling

Browse Source
This commit is contained in:
Dowon
2025-02-18 20:18:42 +09:00
parent 589412052d
commit 06100063b3
2 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/!adetailer.py
View File

@@ -18,8 +18,8 @@ import modules
from aaaaaa.conditional import create_binary_mask, schedulers
from aaaaaa.helper import (
PPImage,
change_torch_load,
copy_extra_params,
disable_safe_unpickle,
pause_total_tqdm,
preserve_prompts,
)
@@ -825,8 +825,8 @@ class AfterDetailerScript(scripts.Script):
pred = mediapipe_predict(args.ad_model, pp.image, args.ad_confidence)
else:
with change_torch_load():
ad_model = self.get_ad_model(args.ad_model)
ad_model = self.get_ad_model(args.ad_model)
with disable_safe_unpickle():
pred = ultralytics_predict(
ad_model,
image=pp.image,