examples/research_projects/rag-end2end-retriever/finetune_rag.py/GenerativeQAModule/forward
class GenerativeQAModule:    def forward(self, input_ids, **kwargs):
        return self.model(input_ids, **kwargs)