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: suppress mediapipe error

Browse Source
This commit is contained in:
Dowon
2024-08-03 12:57:09 +09:00
parent b700b354d5
commit cdb7bd6357
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
adetailer/mediapipe.py
View File

@@ -21,7 +21,10 @@ def mediapipe_predict(
} }
if model_type in mapping: if model_type in mapping:
func = mapping[model_type] func = mapping[model_type]
return func(image, confidence) try:
return func(image, confidence)
except Exception:
return PredictOutput()
msg = f"[-] ADetailer: Invalid mediapipe model type: {model_type}, Available: {list(mapping.keys())!r}" msg = f"[-] ADetailer: Invalid mediapipe model type: {model_type}, Available: {list(mapping.keys())!r}"
raise RuntimeError(msg) raise RuntimeError(msg)