diff --git a/pyproject.toml b/pyproject.toml
index 23ea930f..8e0708f1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,7 +32,7 @@ dependencies = [
   "moviepy>=1.0.3",
   "tensorboard>=2.10",
   "python-dotenv>=1.0.0",
-  "lightning==2.1.*",
+  "lightning>=2.0",
   "lightning-utilities<=0.9",
   "hydra-core==1.3.0",
   "torchmetrics",