def get_inputs(cls, batch, device, model_name):
        """
        Converts a batch of features to model input format,
        used by Transformer.fine_tune.
        """
        batch = tuple(t.to(device) for t in batch)
        inputs = {
            "source_ids": batch[0],
            "target_ids": batch[1],
            "pseudo_ids": batch[2],
            "num_source_tokens": batch[3],
            "num_target_tokens": batch[4],
        }
        return inputs

        Converts a batch of features to model input format,
        used by Transformer.fine_tune.
        
utils_nlp/models/transformers/abstractive_summarization_seq2seq.py