def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--model_name_or_path",
        default="bart-base",
        type=str,
        choices=["bart-base", "bart-large"],
        help="The model name to specify the bart to use. Can be one of ['bart-base', 'bart-large']. ",
    )
    parser.add_argument(
        "--decode_strategy",
        default="sampling",
        type=str,
        choices=["greedy_search", "beam_search", "sampling"],
        help="The decoding strategy. Can be one of ['greedy_search', 'beam_search', 'sampling']",
    )
    parser.add_argument("--num_beams", default=4, type=int, help="The parameters for beam search. ")
    parser.add_argument("--top_k", default=4, type=int, help="The number of candidate to procedure beam search. ")
    parser.add_argument(
        "--top_p", default=1.0, type=float, help="The probability threshold to procedure topp sampling. "
    )
    parser.add_argument("--max_length", default=32, type=int, help="Maximum output length. ")
    parser.add_argument("--use_fp16_decoding", action="store_true", help="Whether to use fp16 decoding to predict. ")
    args = parser.METHOD_NAME()
    return args