immich/machine-learning/app
Mert 87a0ba3db3
feat(ml): export clip models to ONNX and host models on Hugging Face (#4700)
* export clip models

* export to hf

refactored export code

* export mclip, general refactoring

cleanup

* updated conda deps

* do transforms with pillow and numpy, add tokenization config to export, general refactoring

* moved conda dockerfile, re-added poetry

* minor fixes

* updated link

* updated tests

* removed `requirements.txt` from workflow

* fixed mimalloc path

* removed torchvision

* cleaner np typing

* review suggestions

* update default model name

* update test
2023-10-31 05:02:04 -05:00
..
models feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) 2023-10-31 05:02:04 -05:00
__init__.py refactor(ml): modularization and styling (#2835) 2023-06-24 22:18:09 -05:00
config.py docs: remove obsolete environment variables (#3926) 2023-09-01 18:05:45 +07:00
conftest.py feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) 2023-10-31 05:02:04 -05:00
main.py fix(ml): load models in separate threads (#4034) 2023-09-09 16:02:44 +07:00
schemas.py feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) 2023-10-31 05:02:04 -05:00
test_main.py feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) 2023-10-31 05:02:04 -05:00