src/transformers/models/rag/modeling_rag.py/RagTokenForGeneration/set_context_encoder_for_training
class RagTokenForGeneration:    def set_context_encoder_for_training(self, ctx_encoder: PreTrainedModel):
        self.rag.context_encoder_training = True
        self.rag.ctx_encoder = ctx_encoder