examples/research_projects/rag/distributed_ray_retriever.py · e33085d648c9934a88a8b27b509489e724a12ee1 · chenpangpang / transformers

updated the original RAG implementation to be compatible with latest Pytorch-Lightning (#11806) · e33085d6

Shamane Siri authored Jun 09, 2021

* updated the original RAG implementation to be compatible with the latest PL version

* updated the requirements.txt file

* execute make style

* code quality test

* code quality

* conflix resolved in requirement.txt

* code quality

* changed the MyDDP class name to CustomDDP

e33085d6

distributed_ray_retriever.py 7.02 KB

Replace distributed_ray_retriever.py