environment.yaml

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