immich/machine-learning/app/models
Mert c73832bd9c
refactor(ml): model downloading (#3545)
* download facial recognition models

* download hf models

* simplified logic

* updated `predict` for facial recognition

* ensure download method is called

* fixed repo_id for clip

* fixed download destination

* use st's own `snapshot_download`

* conditional download

* fixed predict method

* check if loaded

* minor fixes

* updated mypy overrides

* added pytest-mock

* updated tests

* updated lock
2023-08-05 21:45:13 -05:00
..
__init__.py fixed setting different clip, removed unused stubs (#2987) 2023-06-27 12:21:50 -05:00
base.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
cache.py fix(ml): race condition when loading models (#3207) 2023-07-11 12:01:21 -05:00
clip.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
facial_recognition.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
image_classification.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00