try to fix tests

lucidrains · lucidrains · commit 59c8948c6a8a · 2022-10-29T11:44:17.000-07:00
diff --git 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',
