1 files changed, 5 insertions, 3 deletions

diff --git a/environment.yaml b/environment.yaml
index dfbafaf..85033ce 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -6,6 +6,8 @@ channels:
     - defaults
     - conda-forge
 dependencies:
+    - cuda-nvcc=11.8
+    - cuda-cudart-dev=11.8
     - gcc=11.3.0
     - gxx=11.3.0
     - matplotlib=3.6.2
@@ -20,13 +22,13 @@ dependencies:
           - -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
+          - transformers==4.29.0