src/transformers/models/rag/modeling_rag.py/RagSequenceForGeneration/set_retriever
class RagSequenceForGeneration:    def set_retriever(self, retriever: RagRetriever):
        self.rag.retriever = retriever