From d4ff50ee1466d7f796bb73ad992f93668a0d99b3 Mon Sep 17 00:00:00 2001
From: Koen van Eijk <vaneijk.koen@gmail.com>
Date: Wed, 19 Jun 2024 20:20:03 +0200
Subject: [PATCH] added compatibie version of torchvision

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index c15a2af..0ee34cc 100644
--- a/setup.py
+++ b/setup.py
@@ -12,8 +12,8 @@ install_requires = [
     "numpy==1.26.4",
     "mss==9.0.1",
     "sentence-transformers==3.0.0",
-    "torch==2.2.1",
-    "torchvision==0.18.0",
+    "torch==2.2.0",
+    "torchvision==0.17.0",
     "shapely==2.0.4",
     "h5py==3.11.0",
     "rapidfuzz==3.9.3",