def METHOD_NAME(self, args, model, **kwargs):
    input_ids = rand_int_tensor(1, model.config.vocab_size, [args.batch_size, args.max_seq_len])
    start_positions = rand_int_tensor(
        0,
        args.max_seq_len,
        [
            args.batch_size,
        ],
    )
    end_positions = rand_int_tensor(
        0,
        args.max_seq_len,
        [
            args.batch_size,
        ],
    )
    return {"input_ids": input_ids, "start_positions": start_positions, "end_positions": end_positions}