summaryrefslogtreecommitdiffstats
path: root/environment.yaml
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-03-17 15:18:20 +0100
committerVolpeon <git@volpeon.ink>2023-03-17 15:18:20 +0100
commit8abbd633d8ee7500058b2f1f69a6d6611b5a4450 (patch)
treef60d6e384966ba05354b30f08a32a38279b56165 /environment.yaml
parentUpdate (diff)
downloadtextual-inversion-diff-8abbd633d8ee7500058b2f1f69a6d6611b5a4450.tar.gz
textual-inversion-diff-8abbd633d8ee7500058b2f1f69a6d6611b5a4450.tar.bz2
textual-inversion-diff-8abbd633d8ee7500058b2f1f69a6d6611b5a4450.zip
Test: https://arxiv.org/pdf/2303.09556.pdf
Diffstat (limited to 'environment.yaml')
-rw-r--r--environment.yaml46
1 files changed, 22 insertions, 24 deletions
diff --git a/environment.yaml b/environment.yaml
index 018a9ab..9355f37 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,27 +1,25 @@
1name: ldd 1name: ldd
2channels: 2channels:
3 - pytorch 3 - pytorch
4 - nvidia 4 - nvidia
5 - xformers/label/dev 5 - xformers/label/dev
6 - defaults 6 - defaults
7dependencies: 7dependencies:
8 - cudatoolkit=11.7 8 - matplotlib=3.6.2
9 - libcufile=1.4.0.31 9 - numpy=1.23.4
10 - matplotlib=3.6.2 10 - pip=22.3.1
11 - numpy=1.23.4 11 - python=3.10.8
12 - pip=22.3.1 12 - pytorch=2.0.0=*cuda11.8*
13 - python=3.10.8 13 - torchvision=0.15.0
14 - pytorch=1.13.1=*cuda* 14 # - xformers=0.0.17.dev476
15 - torchvision=0.14.1 15 - pip:
16 - xformers=0.0.17.dev466 16 - -e .
17 - pip: 17 - -e git+https://github.com/huggingface/diffusers#egg=diffusers
18 - -e . 18 - accelerate==0.17.1
19 - -e git+https://github.com/huggingface/diffusers#egg=diffusers 19 - bitsandbytes==0.37.1
20 - accelerate==0.16.0 20 - python-slugify>=6.1.2
21 - bitsandbytes==0.37.0 21 - safetensors==0.3.0
22 - python-slugify>=6.1.2 22 - setuptools==65.6.3
23 - safetensors==0.2.8 23 - test-tube>=0.7.5
24 - setuptools==65.6.3 24 - transformers==4.27.1
25 - test-tube>=0.7.5 25 - triton==2.0.0
26 - transformers==4.26.1
27 - triton==2.0.0