AutoGaze / environment.yaml
bfshi's picture
initial
7e3b296
name: gengaze_demo
channels:
- nvidia
- conda-forge
- defaults
dependencies:
- python=3.10
- pip
- nvidia::cuda-toolkit=12.6
- pip:
- torch==2.7.1
- torchvision==0.22.1
- torchaudio==2.7.1
- numpy==1.26.4
- pillow==10.4.0
- matplotlib==3.10.1
- gradio>=4.0.0
- spaces
- flash_attn==2.8.0.post2
- hydra-core==1.3.2
- wandb==0.21.0
- loguru==0.7.3
- timm==1.0.15
- tqdm==4.67.1
- transformers==4.53.0
- omegaconf==2.3.0
- einops==0.8.1
- av==14.4.0
- imageio==2.37.0