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",