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

name: ldd
channels:
    - pytorch
    - nvidia
    - xformers/label/dev
    - defaults
dependencies:
    - 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.18.dev504
    - pip:
          - -e .
          - -e git+https://github.com/huggingface/diffusers#egg=diffusers
          - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation
          - accelerate==0.17.1
          - bitsandbytes==0.37.2
          - lion-pytorch==0.0.7
          - peft==0.2.0
          - python-slugify>=6.1.2
          - safetensors==0.3.0
          - setuptools==65.6.3
          - test-tube>=0.7.5
          - timm==0.8.17.dev0
          - transformers==4.27.1
          - triton==2.0.0.post1