From 9f150773374a2bffb0bc3ea17b1dae4c59326eab Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Sun, 25 Sep 2022 19:33:18 +0200
Subject: [PATCH] Requirements Updates

---
 environments/rocm.yml | 4 ++--
 requirements.txt      | 4 ++--
 requirements_mtj.txt  | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/environments/rocm.yml b/environments/rocm.yml
index fef38892..e885f4df 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -18,8 +18,8 @@ dependencies:
   - apispec-webframeworks
   - loguru
   - pip:
-    - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
-    - torch==1.10.*
+    - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
+    - torch
     - torchvision
     - flask-cloudflared
     - flask-ngrok
diff --git a/requirements.txt b/requirements.txt
index 4923a140..c0ec6395 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ transformers>=4.20.1
 Flask
 Flask-SocketIO
 requests
-torch==1.11
+torch<=1.12.*
 flask-cloudflared
 flask-ngrok
 eventlet
@@ -15,4 +15,4 @@ accelerate
 flask-session
 marshmallow>=3.13
 apispec-webframeworks
-loguru
\ No newline at end of file
+loguru
diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index 9092aef0..0a915a07 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -1,4 +1,4 @@
-torch >= 1.9, <= 1.11
+torch >= 1.9, <= 1.12.*
 numpy
 tqdm
 requests
@@ -19,4 +19,4 @@ bleach==4.1.0
 flask-session
 marshmallow>=3.13
 apispec-webframeworks
-loguru
\ No newline at end of file
+loguru