diff --git a/requirements.txt b/requirements.txt index 80b47cad..b1e2247c 100644 --- 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 diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 0bd02cda..743c9c1d 100644 --- 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