summaryrefslogtreecommitdiffstats
path: root/environment.yaml
blob: 57e090cb59def297b0205afb1956d57f09c8d24f (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
28
29
30
31
32
33
34
35
36
37
38
name: ldd
channels:
  - pytorch
  - xformers/label/dev
  - defaults
dependencies:
  - cudatoolkit=11.3
  - numpy=1.22.3
  - pip=20.3
  - python=3.9.13
  - pytorch=1.12.1
  - torchvision=0.13.1
  - pandas=1.4.3
  - xformers=0.0.15.dev344
  - 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/devilismyfriend/latent-diffusion#egg=latent-diffusion
      - -e git+https://github.com/huggingface/diffusers#egg=diffusers
      - accelerate==0.12.0
      - albumentations==1.1.0
      - bitsandbytes==0.35.4
      - einops==0.4.1
      - imageio==2.22.0
      - k-diffusion==0.0.12
      - 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.25.1
      - triton==2.0.0.dev20221105