immich/machine-learning/export/env.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
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 06:02:04 -04:00			`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`