feat(ml): export clip models to ONNX and host models on Hugging Face (#4700)

Browse Source

* 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

...

This commit is contained in:

Mert

2023-10-31 06:02:04 -04:00

committed by GitHub

parent 3212a47720

commit 87a0ba3db3

29 changed files with 6192 additions and 2043 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

machine-learning/export/models/__init__.py Normal file