From f0a171923cc8240177302f3dccb6177a2c646ab3 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 11 May 2023 18:37:43 +0200
Subject: Update

---
 environment.yaml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'environment.yaml')

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
-- 
cgit v1.2.3-54-g00ecf