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

name: ldd
channels:
  - pytorch-nightly
  - nvidia
  - xformers/label/dev
  - defaults
dependencies:
  - cudatoolkit=11.7
  - matplotlib=3.6.2
  - numpy=1.23.4
  - pip=22.3.1
  - python=3.10.8
  - pytorch=2.0.0.dev20230216=*cuda*
  - torchvision=0.15.0.dev20230216
  - pip:
      - -e .
      - -e git+https://github.com/huggingface/diffusers#egg=diffusers
      - accelerate==0.16.0
      - bitsandbytes==0.37.0
      - python-slugify>=6.1.2
      - safetensors==0.2.8
      - setuptools==65.6.3
      - test-tube>=0.7.5
      - transformers==4.26.1