diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 18 |
1 files changed, 7 insertions, 11 deletions
diff --git a/environment.yaml b/environment.yaml index f5632bf..8010c09 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -1,28 +1,24 @@ | |||
1 | name: ldd | 1 | name: ldd |
2 | channels: | 2 | channels: |
3 | - pytorch | 3 | - pytorch-nightly |
4 | - nvidia | 4 | - nvidia |
5 | - xformers/label/dev | 5 | - xformers/label/dev |
6 | - defaults | 6 | - defaults |
7 | dependencies: | 7 | dependencies: |
8 | - cudatoolkit=11.3 | 8 | - cudatoolkit=11.7 |
9 | - libcufile=1.4.0.31 | ||
10 | - matplotlib=3.6.2 | 9 | - matplotlib=3.6.2 |
11 | - numpy=1.23.4 | 10 | - numpy=1.23.4 |
12 | - pip=22.3.1 | 11 | - pip=22.3.1 |
13 | - python=3.10.8 | 12 | - python=3.10.8 |
14 | - pytorch=1.13.1=*cuda* | 13 | - pytorch=2.0.0.dev20230216=*cuda* |
15 | - torchvision=0.14.1 | 14 | - torchvision=0.15.0.dev20230216 |
16 | - pip: | 15 | - pip: |
17 | - -e . | 16 | - -e . |
18 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 17 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
19 | - -e git+https://github.com/cloneofsimo/lora#egg=lora-diffusion | 18 | - accelerate==0.16.0 |
20 | - accelerate==0.15.0 | ||
21 | - bitsandbytes==0.37.0 | 19 | - bitsandbytes==0.37.0 |
22 | - python-slugify>=6.1.2 | 20 | - python-slugify>=6.1.2 |
23 | - safetensors==0.2.7 | 21 | - safetensors==0.2.8 |
24 | - setuptools==65.6.3 | 22 | - setuptools==65.6.3 |
25 | - test-tube>=0.7.5 | 23 | - test-tube>=0.7.5 |
26 | - transformers==4.25.1 | 24 | - transformers==4.26.1 |
27 | - triton==2.0.0.dev20221202 | ||
28 | - xformers==0.0.17.dev443 | ||