environment.yml

name: bytedream
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pip
  - pip:
    - transformers>=4.35.0
    - diffusers>=0.24.0
    - torch>=2.1.0
    - torchaudio>=2.1.0
    - accelerate>=0.25.0
    - numpy>=1.24.0
    - pillow>=10.0.0
    - opencv-python>=4.8.0
    - safetensors>=0.4.0
    - huggingface_hub>=0.19.0
    - gradio>=4.0.0
    - tqdm>=4.66.0
    - pyyaml>=6.0
    - matplotlib>=3.8.0
    - scipy>=1.11.0
    - einops>=0.7.0