4 Commits

Author	SHA1	Message	Date
Mert	41580696c7	feat(ml): add more search models (#11468) ... * update export code * add uuid glob, sort model names * add new models to ml, sort names * add new models to server, sort by dims and name * typo in name * update export dependencies * onnx save function * format	2024-07-31 04:34:45 +00:00
Mert	e2666f0e74	fix(ml): remove unused import (#6356) ... * remove unused import * formatting	2024-01-13 00:25:26 -05:00
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