ReNO / environment.yml
fffiloni's picture
Upload 24 files
ca25718 verified
raw
history blame contribute delete
412 Bytes
name: reno
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- python=3.11
- pytorch=2.3
- torchvision=0.18.0
- pytorch-cuda
- pytorch-lightning=2.2
- pip
- pip:
- datasets==2.18
- transformers==4.38.2
- diffusers==0.28
- hpsv2==1.2
- image-reward==1.5
- open-clip-torch==2.24
- blobfile
- openai-clip
- setuptools==60.2
- optimum
- xformers