def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--model_name_or_path", default="bart-base", type=str, help="The model name to specify the bart to use. "
    )
    parser.add_argument(
        "--inference_model_dir", default="./infer_model/", type=str, help="Path to save inference model of bart. "
    )
    parser.add_argument("--topk", default=4, type=int, help="The number of candidate to procedure top_k sampling. ")
    parser.add_argument(
        "--topp", default=1.0, type=float, help="The probability threshold to procedure top_p sampling. "
    )
    parser.add_argument("--max_out_len", default=20, type=int, help="Maximum output length. ")
    parser.add_argument("--temperature", default=1.0, type=float, help="The temperature to set. ")
    parser.add_argument("--num_return_sequences", default=1, type=int, help="The number of returned sequences. ")
    parser.add_argument("--use_fp16_decoding", action="store_true", help="Whether to use fp16 decoding to predict. ")
    parser.add_argument(
        "--decoding_strategy",
        default="beam_search",
        choices=["sampling", "beam_search"],
        type=str,
        help="The main strategy to decode. ",
    )
    parser.add_argument("--num_beams", default=5, type=int, help="The number of candidate to procedure beam search. ")
    parser.add_argument(
        "--diversity_rate", default=0.0, type=float, help="The diversity rate to procedure beam search. "
    )
    parser.add_argument("--repetition_penalty", default=1.0, type=float, help="The repetition_penalty to set. ")
    parser.add_argument("--length_penalty", default=0.0, type=float, help="The length penalty to decode. ")
    parser.add_argument("--early_stopping", action="store_true", help="Whether to do early stopping. ")
    args = parser.METHOD_NAME()
    return args