1 files changed, 9 insertions, 5 deletions

diff --git a/environment.yaml b/environment.yaml
index a95df2a..b161244 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -4,26 +4,30 @@ channels:
     - nvidia
     - xformers/label/dev
     - defaults
+    - conda-forge 
 dependencies:
+    - gcc=11.3.0
+    - gxx=11.3.0
     - matplotlib=3.6.2
     - numpy=1.23.4
     - pip=22.3.1
     - python=3.10.8
     - pytorch=2.0.0=*cuda11.8*
     - torchvision=0.15.0
-    - xformers=0.0.18.dev504
+    - 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
-          - accelerate==0.17.1
-          - bitsandbytes==0.37.2
+          - bitsandbytes==0.38.1
+          - hidet==0.2.3
           - lion-pytorch==0.0.7
           - peft==0.2.0
           - python-slugify>=6.1.2
-          - safetensors==0.3.0
+          - safetensors==0.3.1
           - setuptools==65.6.3
           - test-tube>=0.7.5
           - timm==0.8.17.dev0
-          - transformers==4.27.1
+          - transformers==4.28.1
           - triton==2.0.0.post1