⬆️ Bump upper bound of torch

Signed-off-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com>
caikit · Aug 5, 2024 · c4f86ec · c4f86ec
1 parent 79ac58e
commit c4f86ec
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>=2.3.1,<2.4.0",
     "tokenizers>=0.13.3",
-    "torch>=2.3.1,<2.4.0",
+    "torch>=2.3.1,<2.5.0",
     "tqdm>=4.65.0",
     "transformers>=4.32.0",
     "peft==0.6.0",