diff --git a/setup.py b/setup.py
index 4def612..1553e22 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
     install_requires=[
-        "torch<=2.5.2",
+        "torch<=2.5.1",
         "nnunetv2<=2.5.1",
         "argparse",
         "huggingface_hub"