From 314077b5545acde8ceb2ba98d4255e59abcc9d79 Mon Sep 17 00:00:00 2001
From: Gitanshu Sardana <gitanshusardana@gmail.com>
Date: Fri, 12 Mar 2021 15:04:27 -0800
Subject: [PATCH] Pin torch until #2248 is fixed

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

diff --git a/setup.py b/setup.py
index b4f394a69a..9ccceb6f44 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,7 @@
     'setproctitle>=1.0',
     'tensorflow>=1.14',
     'tensorflow-probability>=0.11.0',
-    'torch>=1.0.0,!=1.5.0',
+    'torch>=1.0.0,!=1.5.0,<1.8.0',
     'torchvision>=0.2.1',
 ]