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