chore: update torch version

Modalities · Feb 10, 2025 · 80b962b · 80b962b
1 parent 3ebaa91
commit 80b962b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ description = "Modalities, a PyTorch-native framework for distributed and reprod
 readme = "README.md"
 dependencies = [
     "numpy<2.0",
-    "torch~=2.4.1",
+    "torch==2.6.0",
     "packaging",
     "tqdm",
     "pyyaml",