From 2abb5190b4f6ccbc51de578995f55ca57b910814 Mon Sep 17 00:00:00 2001
From: Iris Liu <irisliu.0616@gmail.com>
Date: Tue, 16 Mar 2021 11:42:00 -0700
Subject: [PATCH] Pin torchvision version until #2248 is fixed (#2252)

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

diff --git a/setup.py b/setup.py
index 9ccceb6f44..0a1e0ff317 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@
     'tensorflow>=1.14',
     'tensorflow-probability>=0.11.0',
     'torch>=1.0.0,!=1.5.0,<1.8.0',
-    'torchvision>=0.2.1',
+    'torchvision>=0.2.1,<=0.8.2',
 ]
 
 if sys.version_info < (3, 7):