Torch fixes for colab

2022-09-25 19:47:08 +02:00 · 2022-09-25 19:47:08 +02:00 · 50266ab49a
parent b0a32d3646
commit 50266ab49a
2 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -2,7 +2,7 @@ transformers>=4.20.1
 Flask
 Flask-SocketIO
 requests
-torch<=1.12
+torch >= 1.9, < 1.13
 flask-cloudflared
 flask-ngrok
 eventlet
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@ -1,4 +1,4 @@
-torch >= 1.9, <= 1.12
+torch >= 1.9, < 1.13
 numpy
 tqdm
 requests