diff --git a/pyproject.toml b/pyproject.toml
index d0a13f4..ef0a6a8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -37,7 +37,7 @@ dependencies = [
     "supervision~=0.24.0rc1",
     "requests>=2.31.0,<=2.32.3",
     "transformers>=4.44.2,<4.46.0",
-    "torch~=2.4.0",
+    "torch>=2.4,<2.6",
     "accelerate>=0.33,<1.1",
     "sentencepiece~=0.2.0",
     "peft>=0.12,<0.14",