From 177e551da6d14902543be7bf95561a591fb5ae21 Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Tue, 19 Dec 2023 00:13:09 +0100
Subject: [PATCH] ROCm dependency update

---
 environments/rocm.yml | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/environments/rocm.yml b/environments/rocm.yml
index 2a6043ea..87944c7d 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -24,18 +24,18 @@ dependencies:
   - psutil
   - ffmpeg
   - pip:
-    - --extra-index-url https://download.pytorch.org/whl/rocm5.2
-    - torch==1.13.1+rocm5.2
+    - --extra-index-url https://download.pytorch.org/whl/rocm5.6
+    - torch==2.1.2+rocm5.6
     - flask-cloudflared==0.0.10
     - flask-ngrok
     - flask-cors
     - Werkzeug==2.3.7
     - lupa==1.10
-    - transformers[sentencepiece]==4.34.0
-    - huggingface_hub==0.16.4
-    - optimum[onnxruntime]==1.13.2
-    - safetensors==0.3.3
-    - accelerate==0.21.0
+    - transformers[sentencepiece]==4.36.1
+    - huggingface_hub==0.19.4
+    - optimum[onnxruntime]==1.16.1
+    - safetensors==0.4.1
+    - accelerate==0.25.0
     - git+https://github.com/VE-FORBRYDERNE/mkultra
     - ansi2html
     - flask_compress
@@ -45,8 +45,9 @@ dependencies:
     - diffusers
     - git+https://github.com/0cc4m/hf_bleeding_edge/
     - einops
-    - peft==0.3.0
+    - peft==0.7.1
     - windows-curses; sys_platform == 'win32'
     - pynvml
-    - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+rocm5.4.2-cp38-cp38-linux_x86_64.whl
-    - omegaconf
\ No newline at end of file
+    - https://huggingface.github.io/autogptq-index/whl/rocm561/auto-gptq/auto_gptq-0.5.1%2Brocm5.6.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+    - omegaconf
+    - https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11-cp38-cp38-linux_x86_64.whl
\ No newline at end of file