fix: adopted torch version

Modalities · Jul 11, 2024 · 9ff5e6c · 9ff5e6c
1 parent 69d0dd6
commit 9ff5e6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,9 +4,9 @@ version = "0.1.0"
 requires-python = ">=3.10,<3.12"
 description = "Modalities, a python framework for distributed and reproducible foundation model training."
 dependencies = [
+    "torch==2.3.1",
     "mamba-ssm",
     "numpy<2.0",
-    "torch>=2.0",
     "tqdm",
     "pyyaml",
     "transformers",