ashipov/immich
1
0
Fork 1
forked from dark_thunder/immich
Code Pull Requests Activity

feat(ml): add face models (#4952)

Browse Source 
added models to config dropdown

fixed downloading

updated tests

use hf for face models

formatting
This commit is contained in:
Mert
2023-11-11 20:04:49 -05:00
committed by GitHub
parent 7fca0d8da5
commit 328a58ac0d
8 changed files with 101 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
machine-learning/app/config.py
View File

@ -38,8 +38,16 @@ class LogSettings(BaseSettings):
_clean_name = str.maketrans(":\\/", "___", ".")
def clean_name(model_name: str) -> str:
return model_name.split("/")[-1].translate(_clean_name)
def get_cache_dir(model_name: str, model_type: ModelType) -> Path:
return Path(settings.cache_folder) / model_type.value / model_name.translate(_clean_name)
return Path(settings.cache_folder) / model_type.value / clean_name(model_name)
def get_hf_model_name(model_name: str) -> str:
return f"immich-app/{clean_name(model_name)}"
LOG_LEVELS: dict[str, int] = {