Merge pull request #398 from caikit/dependabot/pip/torch-gte-2.3.1-an…

…d-lt-2.6.0 Update torch requirement from <2.5.0,>=2.3.1 to >=2.3.1,<2.6.0
caikit · Oct 30, 2024 · bfd3d4d · bfd3d4d
2 parents 0219d50 + e471a2a
commit bfd3d4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
     "scipy>=1.8.1",
     "sentence-transformers>=3.0.0,<3.1.0",
     "tokenizers>=0.13.3",
-    "torch>=2.3.1,<2.5.0",
+    "torch>=2.3.1,<2.6.0",
     "tqdm>=4.65.0",
     "transformers>=4.38.0,<4.44.0",
     "peft==0.6.0",