diff options
| author | Volpeon <git@volpeon.ink> | 2023-05-16 16:48:51 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-05-16 16:48:51 +0200 |
| commit | 55a12f2c683b2ecfa4fc8b4015462ad2798abda5 (patch) | |
| tree | feeb3f9a041466e773bb5921cbf0adb208d60a49 /environment_nightly.yaml | |
| parent | Avoid model recompilation due to varying prompt lengths (diff) | |
| download | textual-inversion-diff-55a12f2c683b2ecfa4fc8b4015462ad2798abda5.tar.gz textual-inversion-diff-55a12f2c683b2ecfa4fc8b4015462ad2798abda5.tar.bz2 textual-inversion-diff-55a12f2c683b2ecfa4fc8b4015462ad2798abda5.zip | |
Fix LoRA training with DAdan
Diffstat (limited to 'environment_nightly.yaml')
| -rw-r--r-- | environment_nightly.yaml | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/environment_nightly.yaml b/environment_nightly.yaml index 4c5c798..d315bd8 100644 --- a/environment_nightly.yaml +++ b/environment_nightly.yaml | |||
| @@ -4,28 +4,31 @@ channels: | |||
| 4 | - nvidia | 4 | - nvidia |
| 5 | - xformers/label/dev | 5 | - xformers/label/dev |
| 6 | - defaults | 6 | - defaults |
| 7 | - conda-forge | 7 | - conda-forge |
| 8 | dependencies: | 8 | dependencies: |
| 9 | - cuda-nvcc=12.1.105 | 9 | - cuda-nvcc=11.8 |
| 10 | - cuda-cudart-dev=11.8 | ||
| 11 | - gcc=11.3.0 | ||
| 12 | - gxx=11.3.0 | ||
| 10 | - matplotlib=3.6.2 | 13 | - matplotlib=3.6.2 |
| 11 | - numpy=1.24.3 | 14 | - numpy=1.24.3 |
| 12 | - pip=22.3.1 | 15 | - pip=22.3.1 |
| 13 | - python=3.10.8 | 16 | - python=3.10.8 |
| 14 | - pytorch=2.1.0.dev20230429=*cuda12.1* | 17 | - pytorch=2.1.0.dev20230515=*cuda11.8* |
| 15 | - torchvision=0.16.0.dev20230429 | 18 | - torchvision=0.16.0.dev20230516 |
| 16 | # - xformers=0.0.19 | 19 | # - xformers=0.0.19 |
| 17 | - pip: | 20 | - pip: |
| 18 | - -e . | 21 | - -e . |
| 19 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate | 22 | - -e git+https://github.com/huggingface/accelerate#egg=accelerate |
| 20 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 23 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers |
| 21 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation | 24 | - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation |
| 25 | - --pre --extra-index-url https://download.hidet.org/whl hidet | ||
| 22 | - bitsandbytes==0.38.1 | 26 | - bitsandbytes==0.38.1 |
| 23 | - hidet==0.2.3 | ||
| 24 | - lion-pytorch==0.0.7 | 27 | - lion-pytorch==0.0.7 |
| 25 | - peft==0.2.0 | 28 | - peft==0.3.0 |
| 26 | - python-slugify>=6.1.2 | 29 | - python-slugify>=6.1.2 |
| 27 | - safetensors==0.3.1 | 30 | - safetensors==0.3.1 |
| 28 | - setuptools==65.6.3 | 31 | - setuptools==65.6.3 |
| 29 | - test-tube>=0.7.5 | 32 | - test-tube>=0.7.5 |
| 30 | - timm==0.8.17.dev0 | 33 | - timm==0.9.2 |
| 31 | - transformers==4.28.1 | 34 | - transformers==4.29.1 |