Fix-rag-finetune-project-requirement (#21697)

pin pytorch lightning requirement
2023-02-20 17:23:39 +01:00
parent 49ab16239c
commit 4194e5f42b
1 changed files with 1 additions and 1 deletions
--- a/examples/research_projects/rag/requirements.txt
+++ b/examples/research_projects/rag/requirements.txt
@@ -3,6 +3,6 @@ datasets >= 1.0.1
 psutil >= 5.7.0
 torch >= 1.4.0
 ray >= 1.10.0
-pytorch-lightning >= 1.5.10
+pytorch-lightning >= 1.5.10, <=1.6.0
 transformers
 GitPython