diff --git a/requirements.txt b/requirements.txt
index 0d8fa9d..82e1e91 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ tensorboard==2.15.0
 torch==2.2.2
 tqdm>=4.62
 scipy>=1.5
-lightning==2.1.0
+lightning==2.3.3
 
 nltk>=3.6.7
 transformers==4.39.2
diff --git a/setup.py b/setup.py
index 1a2436d..f6b9b31 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
         'protobuf<4',
         'pydantic==1.10.13',
         'truecase==0.0.14',
-        'lightning==2.1.0',
+        'lightning==2.3.3',
         # 'summac',
         'compress_json==1.0.10',
     ],