From c5d7ce8422c353271faed371a2cfcf2cfb912a7b Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Thu, 6 Aug 2020 10:28:57 +0200
Subject: [PATCH] Remove torchvision, fix torch version.

---
 setup.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 328d05d..03c533a 100644
--- a/setup.py
+++ b/setup.py
@@ -10,8 +10,7 @@ REQUIREMENTS = [
     'requests==2.23.0',
     'overrides==3.0.0',
     'tensorboard==2.1.0',
-    'torch>=1.5.0',
-    'torchvision>=0.6.0',
+    'torch>=1.5.0,<1.6.0',
     'tqdm==4.43.0',
     'transformers==2.9.1',
     'urllib3==1.24.2',
-- 
GitLab