def create_parser()

in utils/benchmarking_utils.py [0:0]

• 22 lines of code
• 3 McCabe index (conditional complexity)

def create_parser(is_pixart=False):
    """Creates CLI args parser."""
    parser = argparse.ArgumentParser()

    parser.add_argument("--ckpt", type=str, default="stabilityai/stable-diffusion-xl-base-1.0")
    parser.add_argument("--prompt", type=str, default="ghibli style, a fantasy landscape with castles")
    parser.add_argument("--no_bf16", action="store_true")
    parser.add_argument("--no_sdpa", action="store_true")
    parser.add_argument("--batch_size", type=int, default=1)
    parser.add_argument("--num_inference_steps", type=int, default=30)
    parser.add_argument("--enable_fused_projections", action="store_true")

    if not is_pixart:
        parser.add_argument("--upcast_vae", action="store_true")

    if is_pixart:
        parser.add_argument("--compile_transformer", action="store_true")
    else:
        parser.add_argument("--compile_unet", action="store_true")

    parser.add_argument("--compile_vae", action="store_true")
    parser.add_argument("--compile_mode", type=str, default=None, choices=["reduce-overhead", "max-autotune"])
    parser.add_argument("--change_comp_config", action="store_true")
    parser.add_argument("--do_quant", type=str, default=None)
    parser.add_argument("--tag", type=str, default="")
    parser.add_argument("--device", type=str, choices=["cuda", "cpu"], default="cuda")
    return parser