From 4d804552dffd8142cfaadfcea5b2e45e5734d76c Mon Sep 17 00:00:00 2001 From: Dowon Date: Fri, 8 Mar 2024 22:24:28 +0900 Subject: [PATCH] fix: ad_model typing --- adetailer/__version__.py | 2 +- adetailer/common.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/adetailer/__version__.py b/adetailer/__version__.py index de3c281..a4ea88b 100644 --- a/adetailer/__version__.py +++ b/adetailer/__version__.py @@ -1 +1 @@ -__version__ = "24.3.1-dev.0" +__version__ = "24.3.1-dev.1" diff --git a/adetailer/common.py b/adetailer/common.py index 317d9c6..cf3ad3f 100644 --- a/adetailer/common.py +++ b/adetailer/common.py @@ -44,7 +44,7 @@ def scan_model_dir(path_: str | Path) -> list[Path]: def get_models( model_dir: str | Path, extra_dir: str | Path = "", huggingface: bool = True -) -> OrderedDict[str, str | None]: +) -> OrderedDict[str, str]: model_paths = [*scan_model_dir(model_dir), *scan_model_dir(extra_dir)] models = OrderedDict() @@ -63,10 +63,10 @@ def get_models( ) models.update( { - "mediapipe_face_full": None, - "mediapipe_face_short": None, - "mediapipe_face_mesh": None, - "mediapipe_face_mesh_eyes_only": None, + "mediapipe_face_full": "mediapipe_face_full", + "mediapipe_face_short": "mediapipe_face_short", + "mediapipe_face_mesh": "mediapipe_face_mesh", + "mediapipe_face_mesh_eyes_only": "mediapipe_face_mesh_eyes_only", } )