diff --git a/pyproject.toml b/pyproject.toml
index 6b04115..cef25b7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,7 +39,7 @@ train = [
   "torch-scatter",
   "torch-sparse",
   "torch-cluster",
-  "pytorch-lightning==2.6.1",
+  "pytorch-lightning==2.6.4",
   "jsonargparse[signatures]"
 ]