immich/machine-learning/app
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
..
models refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
__init__.py refactor(ml): modularization and styling (#2835) 2023-06-24 22:18:09 -05:00
config.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
conftest.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
main.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00
schemas.py chore(ml): added testing and github workflow (#2969) 2023-06-27 18:21:33 -05:00
test_main.py refactor(ml): model downloading (#3545) 2023-08-05 21:45:13 -05:00