Put back torchvision<=0.12.0; limit torch<2.1.2

ModECI · Jan 25, 2024 · 678ccf3 · 678ccf3
1 parent e4b3376
commit 678ccf3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -89,8 +89,8 @@ optional =
     Jinja2<3.1
     torchviz
     netron
-    torch>=1.11.0
-    torchvision
+    torch<2.1.2,>=1.11.0
+    torchvision<=0.12.0
     h5py
 
 all_except_psyneulink =