def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--model_name_or_path",
        default="gpt2-en",
        type=str,
        choices=["gpt2-en", "gpt2-medium-en", "gpt2-large-en"],
        help="The model name to specify the bart to use. Can be one of ['gpt2-en', 'gpt2-medium-en', 'gpt2-large-en']. ",
    )
    parser.add_argument(
        "--decode_strategy",
        default="sampling",
        type=str,
        choices=["greedy_search", "sampling"],
        help="The decoding strategy. Can be one of ['greedy_search', 'sampling']",
    )
    parser.add_argument("--top_k", default=4, type=int, help="The number of candidate to procedure beam search. ")
    parser.add_argument("--batch_size", default=4, type=int, help="The size of input batch. ")
    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