environment.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

name: ldd
channels:
    - pytorch
    - nvidia
    - xformers/label/dev
    - defaults
    - conda-forge
dependencies:
    - cuda-nvcc=11.8
    - cuda-cudart-dev=11.8
    - gcc=11.3.0
    - gxx=11.3.0
    - matplotlib=3.6.2
    - numpy=1.23.4
    - pip=22.3.1
    - python=3.10.8
    - pytorch=2.0.0=*cuda11.8*
    - torchvision=0.15.0
    - xformers=0.0.19
    - pip:
          - -e .
          - -e git+https://github.com/huggingface/accelerate#egg=accelerate
          - -e git+https://github.com/huggingface/diffusers#egg=diffusers
          - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation
          - --pre --extra-index-url https://download.hidet.org/whl hidet
          - bitsandbytes==0.38.1
          - lion-pytorch==0.0.7
          - peft==0.3.0
          - python-slugify>=6.1.2
          - safetensors==0.3.1
          - setuptools==65.6.3
          - test-tube>=0.7.5
          - timm==0.8.17.dev0
          - transformers==4.29.0