def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--perf_type",
        default="pd",
        type=str,
        choices=["pd", "pd_faster_fp32", "pd_faster_fp16", "hf"],
        help="The type of perf.  ",
    )
    parser.add_argument(
        "--model_name_or_path",
        default="IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese",
        type=str,
        choices=[
            "IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese",
            "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese",
        ],
        help="The model name to specify the pegasus to use.  ",
    )
    parser.add_argument("--num_beams", default=4, type=int, help="The number of beams to procedure beam search. ")
    parser.add_argument("--batch_size", default=1, type=int, help="The size of input batch. ")
    parser.add_argument("--input_len", default=60, type=int, help="The size of model input. ")
    parser.add_argument("--generate_len", default=20, type=int, help="Length of output . ")
    parser.add_argument("--gpu_id", default=2, type=int, help="The id of GPU . ")
    parser.add_argument(
        "--use_faster", action="store_true", help="Whether to process inference using faster pegasus. "
    )
    parser.add_argument("--use_fp16_decoding", action="store_true", help="Whether to use fp16 decoding to predict. ")
    args = parser.METHOD_NAME()
    return args