diff --git a/requirements.txt b/requirements.txt
index 119babd..2883263 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,4 @@ transformers>=4.0.0,<=5.0.0
 gluonnlp
 seaborn
 mxnet
-pytorch-lightning==0.8.5
+pytorch-lightning==2.4.0