10 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	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	54b2779b79	chore(ml): improved logging (#3918) ... * fixed `minScore` not being set correctly * apply to init * don't send `enabled` * fix eslint warning * added logger * added logging * refinements * enable access log for info level * formatting * merged strings --------- Co-authored-by: Alex <alex.tran1502@gmail.com>	2023-08-30 08:22:01 +00:00
Mert	df26e12db6	fix(ml): minScore not being set correctly (#3916) ... * fixed `minScore` not being set correctly * apply to init * don't send `enabled` * fix eslint warning * better error message	2023-08-30 03:16:00 -05:00
Mert	bcc36d14a1	feat(ml)!: customizable ML settings (#3891) ... * consolidated endpoints, added live configuration * added ml settings to server * added settings dashboard * updated deps, fixed typos * simplified modelconfig updated tests * Added ml setting accordion for admin page updated tests * merge `clipText` and `clipVision` * added face distance setting clarified setting * add clip mode in request, dropdown for face models * polished ml settings updated descriptions * update clip field on error * removed unused import * add description for image classification threshold * pin safetensors for arm wheel updated poetry lock * moved dto * set model type only in ml repository * revert form-data package install use fetch instead of axios * added slotted description with link updated facial recognition description clarified effect of disabling tasks * validation before model load * removed unnecessary getconfig call * added migration * updated api updated api updated api --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>	2023-08-29 08:58:00 -05:00
Mert	165b91b068	feat(ml)!: switch image classification and CLIP models to ONNX (#3809)	2023-08-25 06:28:51 +02: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
Mert	47982641b2	fix(ml): clear model cache on load error (#2951) ... * clear model cache on load error * updated caught exceptions	2023-06-27 16:01:24 -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