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
35
36
37

name: ldd
channels:
  - pytorch
  - defaults
dependencies:
  - cudatoolkit=11.3
  - numpy=1.22.3
  - pip=20.3
  - python=3.8.10
  - pytorch=1.12.1
  - torchvision=0.13.1
  - pandas=1.4.3
  - pip:
      - -e .
      - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
      - -e git+https://github.com/openai/CLIP.git@main#egg=clip
      - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
      - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion
      - -e git+https://github.com/ShivamShrirao/diffusers#egg=diffusers
      - accelerate==0.12.0
      - albumentations==1.1.0
      - bitsandbytes==0.34.0
      - einops==0.4.1
      - imageio==2.22.0
      - kornia==0.6
      - pudb==2019.2
      - omegaconf==2.2.3
      - opencv-python-headless==4.6.0.66
      - python-slugify>=6.1.2
      - pytorch-lightning==1.7.7
      - setuptools==59.5.0
      - test-tube>=0.7.5
      - torch-fidelity==0.3.0
      - torchmetrics==0.9.3
      - transformers==4.22.1
      - triton==2.0.0.dev20220924
      - xformers==0.0.13