src/transformers/models/rag/modeling_rag.py/RagTokenForGeneration/set_output_embeddings
class RagTokenForGeneration:    def set_output_embeddings(self, new_embeddings):
        return self.rag.generator.set_output_embeddings(new_embeddings)