diff options
Diffstat (limited to 'environment.yaml')
-rw-r--r-- | environment.yaml | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/environment.yaml b/environment.yaml index 1a55967..2c81a90 100644 --- a/environment.yaml +++ b/environment.yaml | |||
@@ -14,16 +14,17 @@ dependencies: | |||
14 | - numpy=1.24.3 | 14 | - numpy=1.24.3 |
15 | - pip=22.3.1 | 15 | - pip=22.3.1 |
16 | - python=3.10.8 | 16 | - python=3.10.8 |
17 | - pytorch=2.0.0=*cuda11.8* | 17 | - pytorch=2.0.1=*cuda11.8* |
18 | - torchvision=0.15.0 | 18 | - scipy=1.10.1 |
19 | - xformers=0.0.20.dev528 | 19 | - torchvision=0.15.2 |
20 | - xformers=0.0.21.dev542+git.a205b24 | ||
20 | - pip: | 21 | - pip: |
21 | - -e . | 22 | - -e . |
22 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate | 23 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate |
23 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 24 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
24 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation | 25 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation |
25 | - --pre --extra-index-url https://download.hidet.org/whl hidet | 26 | - --pre --extra-index-url https://download.hidet.org/whl hidet |
26 | - bitsandbytes==0.38.1 | 27 | - bitsandbytes==0.39.1 |
27 | - lion-pytorch==0.0.7 | 28 | - lion-pytorch==0.0.7 |
28 | - peft==0.3.0 | 29 | - peft==0.3.0 |
29 | - python-slugify>=6.1.2 | 30 | - python-slugify>=6.1.2 |
@@ -31,4 +32,4 @@ dependencies: | |||
31 | - setuptools==65.6.3 | 32 | - setuptools==65.6.3 |
32 | - test-tube>=0.7.5 | 33 | - test-tube>=0.7.5 |
33 | - timm==0.9.2 | 34 | - timm==0.9.2 |
34 | - transformers==4.29.1 | 35 | - transformers==4.30.1 |