From 678ccf3837dc8ec870f443d1563c52e8f4e2a5fd Mon Sep 17 00:00:00 2001
From: pgleeson
Date: Thu, 25 Jan 2024 11:00:40 +0000
Subject: [PATCH] Put back torchvision<=0.12.0; limit torch<2.1.2
---
setup.cfg | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/setup.cfg b/setup.cfg
index d426f907..7f398856 100644
--- 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 =