diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/environment.yaml b/environment.yaml index de35645..7aa5312 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -1,6 +1,7 @@ | |||
1 | name: ldd | 1 | name: ldd |
2 | channels: | 2 | channels: |
3 | - pytorch | 3 | - pytorch |
4 | - xformers/label/dev | ||
4 | - defaults | 5 | - defaults |
5 | dependencies: | 6 | dependencies: |
6 | - cudatoolkit=11.3 | 7 | - cudatoolkit=11.3 |
@@ -10,13 +11,14 @@ dependencies: | |||
10 | - pytorch=1.12.1 | 11 | - pytorch=1.12.1 |
11 | - torchvision=0.13.1 | 12 | - torchvision=0.13.1 |
12 | - pandas=1.4.3 | 13 | - pandas=1.4.3 |
14 | - xformers=0.0.14.dev315 | ||
13 | - pip: | 15 | - pip: |
14 | - -e . | 16 | - -e . |
15 | - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers | 17 | - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers |
16 | - -e git+https://github.com/openai/CLIP.git@main#egg=clip | 18 | - -e git+https://github.com/openai/CLIP.git@main#egg=clip |
17 | - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion | 19 | - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion |
18 | - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion | 20 | - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion |
19 | - -e git+https://github.com/ShivamShrirao/diffusers#egg=diffusers | 21 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
20 | - accelerate==0.12.0 | 22 | - accelerate==0.12.0 |
21 | - albumentations==1.1.0 | 23 | - albumentations==1.1.0 |
22 | - bitsandbytes==0.34.0 | 24 | - bitsandbytes==0.34.0 |
@@ -34,4 +36,3 @@ dependencies: | |||
34 | - torchmetrics==0.9.3 | 36 | - torchmetrics==0.9.3 |
35 | - transformers==4.23.1 | 37 | - transformers==4.23.1 |
36 | - triton==2.0.0.dev20220924 | 38 | - triton==2.0.0.dev20220924 |
37 | - xformers==0.0.13 | ||