Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-29 18:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: ad_model typing

Browse Source
This commit is contained in:
Dowon
2024-03-08 22:24:28 +09:00
parent 78ca13ebfa
commit 4d804552df
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
adetailer/__version__.py
View File

@@ -1 +1 @@
__version__ = "24.3.1-dev.0" __version__ = "24.3.1-dev.1"

10
adetailer/common.py
View File

@@ -44,7 +44,7 @@ def scan_model_dir(path_: str | Path) -> list[Path]:
def get_models( def get_models(
model_dir: str | Path, extra_dir: str | Path = "", huggingface: bool = True 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)] model_paths = [*scan_model_dir(model_dir), *scan_model_dir(extra_dir)]
models = OrderedDict() models = OrderedDict()
@@ -63,10 +63,10 @@ def get_models(
) )
models.update( models.update(
{ {
"mediapipe_face_full": None, "mediapipe_face_full": "mediapipe_face_full",
"mediapipe_face_short": None, "mediapipe_face_short": "mediapipe_face_short",
"mediapipe_face_mesh": None, "mediapipe_face_mesh": "mediapipe_face_mesh",
"mediapipe_face_mesh_eyes_only": None, "mediapipe_face_mesh_eyes_only": "mediapipe_face_mesh_eyes_only",
} }
) )