diff --git a/setup.py b/setup.py
index 913f2ee2bcd..0a888cc5618 100644
--- a/setup.py
+++ b/setup.py
@@ -82,7 +82,7 @@
         "packaging>=20.0",
         "psutil",
         "pyyaml",
-        "torch>=2.10,<2.12",
+        "torch==2.9.1",
         "huggingface_hub>=0.21.0",
         "safetensors>=0.4.3",
     ],