diff --git a/pyproject.toml b/pyproject.toml
index 75690ec6..73c38354 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,12 +50,12 @@ classifiers = [
 ]
 dependencies = [
     "ase",
-    "dgl>=2.0.0",
+    "dgl<=2.1.0",
     "pymatgen",
-    "lightning",
+    "lightning<=2.4.0",
     "torch<=2.2.1",
     "pydantic",
-    "torchdata",
+    "torchdata<=0.7.1",
     "boto3",
     "numpy<2.0.0"
 ]