diff --git a/requirements.txt b/requirements.txt
index 765f808..bb0235a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,6 +9,6 @@ requests
 einops
 optuna
 seaborn
-lightning==2.0.3
+lightning==2.3.3
 bgflow @ git+https://github.com/RussellALA/bgflow.git
 nflows @ git+https://github.com/RussellALA/nflows.git