diff --git a/pyproject.toml b/pyproject.toml
index d04e5901..9aedbe7b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,7 +39,7 @@ dependencies = [
   "geobench>=1.0.0",
   "mlflow>=2.12.1",
   # broken due to https://github.com/Lightning-AI/pytorch-lightning/issues/19977
-  "lightning>=2, <=2.2.5",
+  "lightning>=2, <=2.4.0",
   # see issue #64
   "albumentations<=1.4.10"
 ]
diff --git a/requirements/required.txt b/requirements/required.txt
index d425942f..2280e7cd 100644
--- a/requirements/required.txt
+++ b/requirements/required.txt
@@ -10,4 +10,4 @@ lightly==1.4.25
 h5py==3.10.0
 geobench==1.0.0
 mlflow==2.14.3
-lightning==2.2.5
\ No newline at end of file
+lightning==2.4.0
\ No newline at end of file