From 69c794506bdcb1775bb01d36e2339bfe4929c58e Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Tue, 22 Aug 2023 17:48:00 +0200
Subject: [PATCH] HF 4.32

---
 environments/huggingface.yml | 10 +++++-----
 environments/rocm.yml        |  8 ++++----
 requirements.txt             |  7 ++++---
 3 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 004c7ecc..a1a788d8 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -32,10 +32,10 @@ dependencies:
     - flask-ngrok
     - flask-cors
     - lupa==1.10
-    - transformers==4.31.0
-    - huggingface_hub==0.15.1
-    - safetensors==0.3.1
-    - accelerate==0.20.3
+    - transformers==4.32.*
+    - huggingface_hub==0.16.4
+    - git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
+    - accelerate==0.21.0
     - git+https://github.com/VE-FORBRYDERNE/mkultra
     - flask-session
     - ansi2html
@@ -50,7 +50,7 @@ dependencies:
     - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux'
     - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-win_amd64.whl; sys_platform == 'win32'
     - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.1/auto_gptq-0.4.1+cu118-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux'
-    - https://github.com/henk717/KoboldAI/releases/download/Snapshot-11-08-23/auto_gptq-0.4.1+cu118-cp38-cp38-win_amd64.whl; sys_platform == 'win32'
+    - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.1/auto_gptq-0.4.1+cu118-cp38-cp38-win_amd64.whl; sys_platform == 'win32'
     - einops
     - peft==0.3.0
     - scipy
diff --git a/environments/rocm.yml b/environments/rocm.yml
index 9538a615..35f29c3e 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -30,10 +30,10 @@ dependencies:
     - flask-ngrok
     - flask-cors
     - lupa==1.10
-    - transformers==4.31.0
-    - huggingface_hub==0.15.1
-    - safetensors==0.3.1
-    - accelerate==0.20.3
+    - transformers==4.32.*
+    - huggingface_hub==0.16.4
+    - git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
+    - accelerate==0.21.0
     - git+https://github.com/VE-FORBRYDERNE/mkultra
     - ansi2html
     - flask_compress
diff --git a/requirements.txt b/requirements.txt
index b90db7d9..f3b2ac94 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
-transformers==4.31.*
-huggingface_hub==0.15.1
+transformers==4.32.*
+huggingface_hub==0.16.4
+git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
 Flask==2.2.3
 Flask-SocketIO==5.3.2
 python-socketio==5.7.2
@@ -15,7 +16,7 @@ markdown
 bleach==4.1.0
 sentencepiece
 protobuf
-accelerate==0.20.3
+accelerate==0.21.0
 flask-session==0.4.0
 marshmallow>=3.13
 apispec-webframeworks