examples/research_projects/seq2seq-distillation/finetune.py/SummarizationModule/forward
class SummarizationModule:    def forward(self, input_ids, **kwargs):
        return self.model(input_ids, **kwargs)