DWongrade torch.

materialsvirtuallab · Aug 13, 2024 · 57e22a1 · 57e22a1
1 parent a268705
commit 57e22a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,7 +53,7 @@ dependencies = [
     "dgl>=2.0.0",
     "pymatgen",
     "lightning",
-    "torch",
+    "torch<=2.2.1",
     "pydantic",
     "torchdata",
     "boto3",