87a0ba3db3
* 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
25 lines
403 B
YAML
25 lines
403 B
YAML
name: base
|
|
channels:
|
|
- conda-forge
|
|
- nvidia
|
|
- pytorch-nightly
|
|
platforms:
|
|
- linux-64
|
|
dependencies:
|
|
- cxx-compiler
|
|
- onnx==1.*
|
|
- onnxruntime==1.*
|
|
- open-clip-torch==2.*
|
|
- orjson==3.*
|
|
- pip
|
|
- python==3.11.*
|
|
- pytorch
|
|
- rich==13.*
|
|
- safetensors==0.*
|
|
- setuptools==68.*
|
|
- torchvision
|
|
- transformers==4.*
|
|
- pip:
|
|
- multilingual-clip
|
|
- onnx-simplifier
|
|
category: main
|