Update pyproject.toml (#369)

clefourrier · web-flow · commit 5f22e34520a3 · 2024-10-18T13:55:56.000+02:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -56,7 +56,7 @@ dependencies = [
     # Base dependencies
     "transformers>=4.38.0",
     "huggingface_hub>=0.23.0",
-    "torch>=2.0",
+    "torch>=2.0,<2.5",
     "GitPython>=3.1.41", # for logging
     "datasets>=2.14.0",
     # Prettiness
