From 50266ab49a8aa96bcea4e7002fc68c131c8184d5 Mon Sep 17 00:00:00 2001 From: Henk Date: Sun, 25 Sep 2022 19:47:08 +0200 Subject: [PATCH] Torch fixes for colab --- requirements.txt | 2 +- requirements_mtj.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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