summaryrefslogtreecommitdiffstats
path: root/environment.yaml
blob: 42b568fa181c4872ceef071d2fa6e4bf6a847da6 (plain) (blame)
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
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
    - 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.1
          - peft==0.2.0
          - python-slugify>=6.1.2
          - safetensors==0.3.0
          - setuptools==65.6.3
          - test-tube>=0.7.5
          - transformers==4.27.1
          - triton==2.0.0
          - xformers==0.0.17.dev480