diff --git a/pyproject.toml b/pyproject.toml
index cf927db..c90de69 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ dependencies = [
   "ftfy==6.1.1",
   "opencv-python==4.8.1.78",
   "einops==0.7.0",
-  "pytorch-lightning==1.9.0",
+  "pytorch-lightning==2.4.0",
   "bitsandbytes==0.44.0",
   "prodigyopt==1.0",
   "lion-pytorch==0.0.6",