immich/machine-learning/export/env.yaml

name: base
channels:
  - conda-forge
  - nvidia
  - pytorch
platforms:
  - linux-64
dependencies:
  - cxx-compiler
  - onnx==1.*
  - onnxruntime==1.*
  - open-clip-torch==2.*
  - orjson==3.*
  - pip
  - python==3.11.*
  - pytorch>=2.3
  - rich==13.*
  - safetensors==0.*
  - setuptools==68.*
  - torchvision
  - transformers==4.*
  - pip:
    - multilingual-clip
    - onnxsim
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 10:02:04 +00:00			`name: base`
			`channels:`
			`- conda-forge`
			`- nvidia`
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			`- pytorch`
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 10:02:04 +00:00			`platforms:`
			`- linux-64`
			`dependencies:`
			`- cxx-compiler`
			`- onnx==1.*`
			`- onnxruntime==1.*`
			`- open-clip-torch==2.*`
			`- orjson==3.*`
			`- pip`
			`- python==3.11.*`
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			`- pytorch>=2.3`
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 10:02:04 +00:00			`- rich==13.*`
			`- safetensors==0.*`
			`- setuptools==68.*`
			`- torchvision`
			`- transformers==4.*`
			`- pip:`
			`- multilingual-clip`
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			`- onnxsim`
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 10:02:04 +00:00			`category: main`