Merge pull request #230 from ndif-team/update-torch-version

Change base torch version to 2.4
ndif-team · Sep 3, 2024 · 3feab84 · 3feab84
2 parents 77e71d7 + 3d1fa98
commit 3feab84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
     "python-socketio[client]",
     "tokenizers>=0.13.0",
     "pydantic>=2.4.0",
-    "torch>=2.1.0",
+    "torch>=2.4.0",
     "sentencepiece",
     "torchvision",
     "accelerate",