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: BREAKING!! model name. add hand model

Browse Source
This commit is contained in:
Bingsu
2023-04-30 16:34:47 +09:00
parent 0970331875
commit 89b1f6a8ae
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
adetailer/common.py
View File

@@ -26,10 +26,11 @@ def get_models(model_dir: str | Path) -> OrderedDict[str, str | None]:
models = OrderedDict( models = OrderedDict(
{ {
"face_yolo8n.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt"), "face_yolov8n.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt"),
"face_yolo8s.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8s.pt"), "face_yolov8s.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8s.pt"),
"mediapipe_face_full": None, "mediapipe_face_full": None,
"mediapipe_face_short": None, "mediapipe_face_short": None,
"hand_yolov8n.pt": hf_hub_download("Bingsu/adetailer", "hand_yolov8n.pt"),
} }
) )