From 59c8948c6a8a5da851f23c6da2070623426ff6a6 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Sat, 29 Oct 2022 11:44:17 -0700
Subject: [PATCH] try to fix tests

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 0877f90..e635fec 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,9 @@
     'pytest-runner',
   ],
   tests_require=[
-    'pytest'
+    'pytest',
+    'torch==1.12.1',
+    'torchvision==0.13.1'
   ],
   classifiers=[
     'Development Status :: 4 - Beta',