diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/environment.yaml b/environment.yaml index a95df2a..b161244 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -4,26 +4,30 @@ channels: | |||
4 | - nvidia | 4 | - nvidia |
5 | - xformers/label/dev | 5 | - xformers/label/dev |
6 | - defaults | 6 | - defaults |
7 | - conda-forge | ||
7 | dependencies: | 8 | dependencies: |
9 | - gcc=11.3.0 | ||
10 | - gxx=11.3.0 | ||
8 | - matplotlib=3.6.2 | 11 | - matplotlib=3.6.2 |
9 | - numpy=1.23.4 | 12 | - numpy=1.23.4 |
10 | - pip=22.3.1 | 13 | - pip=22.3.1 |
11 | - python=3.10.8 | 14 | - python=3.10.8 |
12 | - pytorch=2.0.0=*cuda11.8* | 15 | - pytorch=2.0.0=*cuda11.8* |
13 | - torchvision=0.15.0 | 16 | - torchvision=0.15.0 |
14 | - xformers=0.0.18.dev504 | 17 | - xformers=0.0.19 |
15 | - pip: | 18 | - pip: |
16 | - -e . | 19 | - -e . |
20 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate | ||
17 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 21 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
18 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation | 22 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation |
19 | - accelerate==0.17.1 | 23 | - bitsandbytes==0.38.1 |
20 | - bitsandbytes==0.37.2 | 24 | - hidet==0.2.3 |
21 | - lion-pytorch==0.0.7 | 25 | - lion-pytorch==0.0.7 |
22 | - peft==0.2.0 | 26 | - peft==0.2.0 |
23 | - python-slugify>=6.1.2 | 27 | - python-slugify>=6.1.2 |
24 | - safetensors==0.3.0 | 28 | - safetensors==0.3.1 |
25 | - setuptools==65.6.3 | 29 | - setuptools==65.6.3 |
26 | - test-tube>=0.7.5 | 30 | - test-tube>=0.7.5 |
27 | - timm==0.8.17.dev0 | 31 | - timm==0.8.17.dev0 |
28 | - transformers==4.27.1 | 32 | - transformers==4.28.1 |
29 | - triton==2.0.0.post1 | 33 | - triton==2.0.0.post1 |