Requirements Updates

2022-09-25 19:33:18 +02:00 · 2022-09-25 19:33:18 +02:00 · 9f15077337
parent 6009586ac8
commit 9f15077337
3 changed files with 6 additions and 6 deletions
--- 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
--- 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
+loguru
--- 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
+loguru