4 commits

Author	SHA1	Message	Date
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
Mert	258b98c262	fix(ml): load models in separate threads (#4034) ... * load models in thread * set clip mode logs to debug level * updated tests * made fixtures slightly less ugly * moved responses to json file * formatting	2023-09-09 16:02:44 +07:00
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
Mert	df1e8679d9	chore(ml): added testing and github workflow (#2969) ... * added testing * github action for python, made mypy happy * formatted with black * minor fixes and styling * test model cache * cache test dependencies * narrowed model cache tests * moved endpoint tests to their own class * cleaned up fixtures * formatting * removed unused dep	2023-06-27 18:21:33 -05:00