From 33745669dd8a08d3c5d743de67c0ed169a5d1dce Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Fri, 5 May 2023 13:14:58 +0200
Subject: [PATCH] Pytorch 2.0

---
 environments/huggingface.yml | 6 ++++--
 requirements.txt             | 5 +++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 669d5d6d..b8d640fb 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -1,6 +1,7 @@
 name: koboldai
 channels:
   - pytorch
+  - nvidia
   - conda-forge
   - defaults
 dependencies:
@@ -9,9 +10,9 @@ dependencies:
   - flask-socketio=5.3.2
   - flask-session=0.4.0
   - python-socketio=5.7.2
-  - pytorch=1.11.*
+  - pytorch=2.0.*
   - python=3.8.*
-  - cudatoolkit=11.1
+  - pytorch-cuda=11.8
   - eventlet=0.33.3
   - dnspython=2.2.1
   - markdown
@@ -45,3 +46,4 @@ dependencies:
     - ftfy
     - pydub
     - diffusers
+    - peft
diff --git a/requirements.txt b/requirements.txt
index 6407303c..584e7377 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ Flask==2.2.3
 Flask-SocketIO==5.3.2
 python-socketio==5.7.2
 requests
-torch >= 1.9, < 1.13
+torch == 2.0.*
 flask-cloudflared==0.0.10
 flask-ngrok
 flask-cors
@@ -36,4 +36,5 @@ pytest==7.2.2
 pytest-html==3.2.0
 pytest-metadata==2.0.4
 requests-mock==1.10.0
-safetensors
\ No newline at end of file
+safetensors
+peft
\ No newline at end of file