summaryrefslogtreecommitdiffstats
path: root/environment.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'environment.yaml')
-rw-r--r--environment.yaml11
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