def METHOD_NAME(cls, parser: argparse.ArgumentParser):
    parser.add_argument(
        "--cerebras.device", type=str, help="Device to load model", default="cuda"
    )
    parser.add_argument(
        "--cerebras.max_length",
        type=int,
        default=50,
        help="The maximum length (in tokens) of the generated text.",
    )
    parser.add_argument(
        "--cerebras.do_sample",
        action="store_true",
        default=False,
        help="Whether to use sampling or not (if not, uses greedy decoding).",
    )
    parser.add_argument(
        "--cerebras.no_repeat_ngram_size",
        type=int,
        default=2,
        help="The size of the n-grams to avoid repeating in the generated text.",
    )