def METHOD_NAME(self, model, inputs, return_outputs=False):
    return model(
        input_ids=inputs["input_ids"],
        attention_mask=torch.ones_like(inputs["input_ids"]).bool(),
        labels=inputs["input_ids"],
    ).loss