transformers/examples/research_projects/rag-end2end-retriever/requirements.txt
Stas Bekman fcb0f74397
[research_projects] deal with security alerts (#15594)
* [research_projects] deal with security alerts

* add a note of the original PL ver and warning
2022-02-11 14:31:09 -05:00

8 lines
125 B
Plaintext

faiss-cpu >= 1.7.0
datasets >= 1.6.2
psutil >= 5.7.0
torch >= 1.4.0
pytorch-lightning
nvidia-ml-py3 == 7.352.0
ray >= 1.3.0