diff --git a/setup.py b/setup.py
index 42911d34..eafacfef 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
         'local-attention==1.8.6',
         'pandas==2.0.2',
         'prefigure==0.0.9',
-        'pytorch_lightning==2.1.0',
+        'pytorch_lightning==2.4.0',
         'PyWavelets==1.4.1',
         'safetensors',
         'sentencepiece==0.1.99',