diff --git a/dpkd/transformers/examples/research_projects/rag-end2end-retriever/requirements.txt b/dpkd/transformers/examples/research_projects/rag-end2end-retriever/requirements.txt
index 32025229..563cc8a7 100644
--- a/dpkd/transformers/examples/research_projects/rag-end2end-retriever/requirements.txt
+++ b/dpkd/transformers/examples/research_projects/rag-end2end-retriever/requirements.txt
@@ -2,6 +2,6 @@ faiss-cpu >= 1.7.2
 datasets 
 psutil >= 5.9.1
 torch >= 1.11.0
-pytorch-lightning == 1.6.4
+pytorch-lightning == 2.6.1
 nvidia-ml-py3 == 7.352.0
 ray >=  1.13.0
\ No newline at end of file