7 commits

Author	SHA1	Message	Date
Mert	935f471ccb	chore(ml): use strict mypy (#5001) ... * improved typing * improved export typing * strict mypy & check export folder * formatting * add formatting checks for export folder * re-added init call	2023-11-13 10:18:46 -06:00
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	165b91b068	feat(ml)!: switch image classification and CLIP models to ONNX (#3809)	2023-08-25 06:28:51 +02:00
Mert	848ba685eb	fix(ml): race condition when loading models (#3207) ... * sync model loading, disabled model ttl by default * disable revalidation if model unloading disabled * moved lock	2023-07-11 12:01:21 -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
Mert	a2f5674bbb	refactor(ml): modularization and styling (#2835) ... * basic refactor and styling * removed batching * module entrypoint * removed unused imports * model superclass, model cache now in app state * fixed cache dir and enforced abstract method --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>	2023-06-24 22:18:09 -05:00

Renamed from machine-learning/app/cache.py (Browse further)