diff --git a/setup.py b/setup.py
index d893f9d..1624f9e 100644
--- a/setup.py
+++ b/setup.py
@@ -16,6 +16,6 @@
                       'torch==2.4',
                       'torchaudio==2.4',
                       'stable-audio-tools==0.0.17',
-                      'pytorch-lightning==2.1',
+                      'pytorch-lightning==2.4.0',
                       'prefigure==0.0.9']
 )
\ No newline at end of file