From 55a12f2c683b2ecfa4fc8b4015462ad2798abda5 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 16 May 2023 16:48:51 +0200
Subject: Fix LoRA training with DAdan

---
 environment_nightly.yaml | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

(limited to 'environment_nightly.yaml')

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:
     - nvidia
     - xformers/label/dev
     - defaults
-    - conda-forge 
+    - conda-forge
 dependencies:
-    - cuda-nvcc=12.1.105
+    - cuda-nvcc=11.8
+    - cuda-cudart-dev=11.8
+    - gcc=11.3.0
+    - gxx=11.3.0
     - matplotlib=3.6.2
     - numpy=1.24.3
     - pip=22.3.1
     - python=3.10.8
-    - pytorch=2.1.0.dev20230429=*cuda12.1*
-    - torchvision=0.16.0.dev20230429
+    - pytorch=2.1.0.dev20230515=*cuda11.8*
+    - torchvision=0.16.0.dev20230516
     # - xformers=0.0.19
     - pip:
           - -e .
           - -e git+https://github.com/huggingface/accelerate#egg=accelerate
           - -e git+https://github.com/huggingface/diffusers#egg=diffusers
           - -e git+https://github.com/facebookresearch/dadaptation#egg=dadaptation
+          - --pre --extra-index-url https://download.hidet.org/whl hidet
           - bitsandbytes==0.38.1
-          - hidet==0.2.3
           - lion-pytorch==0.0.7
-          - peft==0.2.0
+          - peft==0.3.0
           - python-slugify>=6.1.2
           - safetensors==0.3.1
           - setuptools==65.6.3
           - test-tube>=0.7.5
-          - timm==0.8.17.dev0
-          - transformers==4.28.1
+          - timm==0.9.2
+          - transformers==4.29.1
-- 
cgit v1.2.3-54-g00ecf