diff options
Diffstat (limited to 'environment_nightly.yaml')
-rw-r--r-- | environment_nightly.yaml | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/environment_nightly.yaml b/environment_nightly.yaml index 4c5c798..d315bd8 100644 --- a/environment_nightly.yaml +++ b/environment_nightly.yaml | |||
@@ -4,28 +4,31 @@ channels: | |||
4 | - nvidia | 4 | - nvidia |
5 | - xformers/label/dev | 5 | - xformers/label/dev |
6 | - defaults | 6 | - defaults |
7 | - conda-forge | 7 | - conda-forge |
8 | dependencies: | 8 | dependencies: |
9 | - cuda-nvcc=12.1.105 | 9 | - cuda-nvcc=11.8 |
10 | - cuda-cudart-dev=11.8 | ||
11 | - gcc=11.3.0 | ||
12 | - gxx=11.3.0 | ||
10 | - matplotlib=3.6.2 | 13 | - matplotlib=3.6.2 |
11 | - numpy=1.24.3 | 14 | - numpy=1.24.3 |
12 | - pip=22.3.1 | 15 | - pip=22.3.1 |
13 | - python=3.10.8 | 16 | - python=3.10.8 |
14 | - pytorch=2.1.0.dev20230429=*cuda12.1* | 17 | - pytorch=2.1.0.dev20230515=*cuda11.8* |
15 | - torchvision=0.16.0.dev20230429 | 18 | - torchvision=0.16.0.dev20230516 |
16 | # - xformers=0.0.19 | 19 | # - xformers=0.0.19 |
17 | - pip: | 20 | - pip: |
18 | - -e . | 21 | - -e . |
19 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate | 22 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate |
20 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 23 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
21 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation | 24 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation |
25 | - --pre --extra-index-url https://download.hidet.org/whl hidet | ||
22 | - bitsandbytes==0.38.1 | 26 | - bitsandbytes==0.38.1 |
23 | - hidet==0.2.3 | ||
24 | - lion-pytorch==0.0.7 | 27 | - lion-pytorch==0.0.7 |
25 | - peft==0.2.0 | 28 | - peft==0.3.0 |
26 | - python-slugify>=6.1.2 | 29 | - python-slugify>=6.1.2 |
27 | - safetensors==0.3.1 | 30 | - safetensors==0.3.1 |
28 | - setuptools==65.6.3 | 31 | - setuptools==65.6.3 |
29 | - test-tube>=0.7.5 | 32 | - test-tube>=0.7.5 |
30 | - timm==0.8.17.dev0 | 33 | - timm==0.9.2 |
31 | - transformers==4.28.1 | 34 | - transformers==4.29.1 |