diff --git a/setup.py b/setup.py
index 89e18cab62..7a8cc898e0 100644
--- a/setup.py
+++ b/setup.py
@@ -130,7 +130,7 @@ _deps = [
     "regex!=2019.12.17",
     "requests",
     "tensorboard",
-    "torch>=1.4",
+    "torch>=1.4,<2.5.0",
     "torchvision",
     "transformers>=4.41.2",
     "urllib3<=2.0.0",