From 392e0552d7d66e54b87e37528fc02edf4b5b98c4 Mon Sep 17 00:00:00 2001 From: Koen van Eijk Date: Wed, 19 Jun 2024 20:01:22 +0200 Subject: [PATCH] Downgraded PyTorch to 2.2.1 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index a058516..c15a2af 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ install_requires = [ "numpy==1.26.4", "mss==9.0.1", "sentence-transformers==3.0.0", - "torch==2.3.0", + "torch==2.2.1", "torchvision==0.18.0", "shapely==2.0.4", "h5py==3.11.0",