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", } )