From 89b1f6a8aee1e6bd0f8e33640855831263e14586 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Sun, 30 Apr 2023 16:34:47 +0900 Subject: [PATCH] fix: BREAKING!! model name. add hand model --- adetailer/common.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/adetailer/common.py b/adetailer/common.py index 89df224..fa84176 100644 --- a/adetailer/common.py +++ b/adetailer/common.py @@ -26,10 +26,11 @@ def get_models(model_dir: str | Path) -> OrderedDict[str, str | None]: models = OrderedDict( { - "face_yolo8n.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt"), - "face_yolo8s.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8s.pt"), + "face_yolov8n.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt"), + "face_yolov8s.pt": hf_hub_download("Bingsu/adetailer", "face_yolov8s.pt"), "mediapipe_face_full": None, "mediapipe_face_short": None, + "hand_yolov8n.pt": hf_hub_download("Bingsu/adetailer", "hand_yolov8n.pt"), } )