diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt
index 01d5521..4a03b81 100644
--- a/requirements/requirements_lightning.txt
+++ b/requirements/requirements_lightning.txt
@@ -1,3 +1,3 @@
-pytorch-lightning==2.3.3
+pytorch-lightning==2.4.0
 lightning==2.3.3 #need to align with nemo pytorch lightning version
 mlflow==2.20.2