Using RAG with local documents

saichandra
Tue, 06 Apr 2021 12:02:26 +0000

Hi,
I have a requirement that model should search for relevant documents to answer the query and I found RAG from Facebook AI which perfectly fits my usecase. I also found this post in which HuggingFace explains RAG and came to know that HF implemented RAG which is awesome!
My doubt is whether I could extend this functionality so that the model should do retrieval from local documents rather than from HF’s wikipedia corpus.
Are there any notebooks to refer to?