WIP

datamol-io · Oct 28, 2023 · d2464fd · d2464fd
1 parent 0e5a22a
commit d2464fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,7 +37,7 @@ dependencies = [
     "universal_pathlib",
     "datamol",
     "numpy",
-    "pytorch>=2.0",
+    "torch>=2.0",
     "transformers",
     "datasets",
     "tokenizers",