diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-17 15:56:54 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-17 15:56:54 +0100 |
| commit | c927eeb3516b8ecae201441a68956f32247a6d7c (patch) | |
| tree | b24e4246a067415151039484d1f8da2658687ef3 /environment.yaml | |
| parent | Remove xformers, switch to Pytorch Nightly (diff) | |
| download | textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.tar.gz textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.tar.bz2 textual-inversion-diff-c927eeb3516b8ecae201441a68956f32247a6d7c.zip | |
Back to xformers
Diffstat (limited to 'environment.yaml')
| -rw-r--r-- | environment.yaml | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/environment.yaml b/environment.yaml index 8010c09..325644f 100644 --- a/environment.yaml +++ b/environment.yaml | |||
| @@ -1,17 +1,18 @@ | |||
| 1 | name: ldd | 1 | name: ldd |
| 2 | channels: | 2 | channels: |
| 3 | - pytorch-nightly | 3 | - pytorch |
| 4 | - nvidia | 4 | - nvidia |
| 5 | - xformers/label/dev | 5 | - xformers/label/dev |
| 6 | - defaults | 6 | - defaults |
| 7 | dependencies: | 7 | dependencies: |
| 8 | - cudatoolkit=11.7 | 8 | - cudatoolkit=11.7 |
| 9 | - libcufile=1.4.0.31 | ||
| 9 | - matplotlib=3.6.2 | 10 | - matplotlib=3.6.2 |
| 10 | - numpy=1.23.4 | 11 | - numpy=1.23.4 |
| 11 | - pip=22.3.1 | 12 | - pip=22.3.1 |
| 12 | - python=3.10.8 | 13 | - python=3.10.8 |
| 13 | - pytorch=2.0.0.dev20230216=*cuda* | 14 | - pytorch=1.13.1=*cuda* |
| 14 | - torchvision=0.15.0.dev20230216 | 15 | - torchvision=0.14.1 |
| 15 | - pip: | 16 | - pip: |
| 16 | - -e . | 17 | - -e . |
| 17 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 18 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
| @@ -22,3 +23,5 @@ dependencies: | |||
| 22 | - setuptools==65.6.3 | 23 | - setuptools==65.6.3 |
| 23 | - test-tube>=0.7.5 | 24 | - test-tube>=0.7.5 |
| 24 | - transformers==4.26.1 | 25 | - transformers==4.26.1 |
| 26 | - triton==2.0.0a2 | ||
| 27 | - xformers==0.0.17.dev451 | ||