diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 34 |
1 files changed, 10 insertions, 24 deletions
diff --git a/environment.yaml b/environment.yaml index 57e090c..e598f72 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -5,34 +5,20 @@ channels: | |||
5 | - defaults | 5 | - defaults |
6 | dependencies: | 6 | dependencies: |
7 | - cudatoolkit=11.3 | 7 | - cudatoolkit=11.3 |
8 | - numpy=1.22.3 | 8 | - numpy=1.23.4 |
9 | - pip=20.3 | 9 | - pip=22.3.1 |
10 | - python=3.9.13 | 10 | - python=3.9.15 |
11 | - pytorch=1.12.1 | 11 | - pytorch=1.13.1 |
12 | - torchvision=0.13.1 | 12 | - torchvision=0.14.1 |
13 | - pandas=1.4.3 | ||
14 | - xformers=0.0.15.dev344 | ||
15 | - pip: | 13 | - pip: |
16 | - -e . | 14 | - -e . |
17 | - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers | ||
18 | - -e git+https://github.com/openai/CLIP.git@main#egg=clip | ||
19 | - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion | ||
20 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 15 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
21 | - accelerate==0.12.0 | 16 | - -e git+https://github.com/cloneofsimo/lora#egg=lora-diffusion |
22 | - albumentations==1.1.0 | 17 | - accelerate==0.15.0 |
23 | - bitsandbytes==0.35.4 | 18 | - bitsandbytes==0.35.4 |
24 | - einops==0.4.1 | ||
25 | - imageio==2.22.0 | ||
26 | - k-diffusion==0.0.12 | ||
27 | - kornia==0.6 | ||
28 | - pudb==2019.2 | ||
29 | - omegaconf==2.2.3 | ||
30 | - opencv-python-headless==4.6.0.66 | ||
31 | - python-slugify>=6.1.2 | 19 | - python-slugify>=6.1.2 |
32 | - pytorch-lightning==1.7.7 | 20 | - setuptools==65.6.3 |
33 | - setuptools==59.5.0 | ||
34 | - test-tube>=0.7.5 | 21 | - test-tube>=0.7.5 |
35 | - torch-fidelity==0.3.0 | ||
36 | - torchmetrics==0.9.3 | ||
37 | - transformers==4.25.1 | 22 | - transformers==4.25.1 |
38 | - triton==2.0.0.dev20221105 | 23 | - triton==2.0.0.dev20221202 |
24 | - xformers==0.0.16rc391 | ||