From 06a62f03bc814f420fd79e1b4ac2e0319dc29063 Mon Sep 17 00:00:00 2001
From: Jesse Swanson <js11133Wnyu.edu>
Date: Fri, 16 Apr 2021 22:35:43 -0400
Subject: [PATCH] update torch to 1.8.1 for test matrix

---
 requirements.txt | 4 ++--
 setup.py         | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 99a76a076..b1bcf36ee 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 -r requirements-no-torch.txt
-torch>=1.5.0
-torchvision==0.6.0
+torch>=1.8.1
+torchvision==0.9.1
diff --git a/setup.py b/setup.py
index c151cea58..1e426a3ee 100644
--- a/setup.py
+++ b/setup.py
@@ -73,10 +73,10 @@
         "scipy == 1.4.1",
         "sentencepiece == 0.1.86",
         "tokenizers == 0.8.1.rc2",
-        "torch >= 1.5.0",
+        "torch >= 1.8.1",
         "tqdm == 4.46.0",
         "transformers == 3.1.0",
-        "torchvision == 0.6.0",
+        "torchvision == 0.9.1",
     ],
     extras_require=extras,
     python_requires=">=3.6.0",