utils.py

import argparse


def get_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-p",
        "--precision",
        type=str,
        default="int4",
        choices=["int4", "bf16"],
        help="Which precisions to use",
    )
    parser.add_argument(
        "--count-use", action="store_true", help="Whether to count the number of uses"
    )
    parser.add_argument("--gradio-root-path", type=str, default="")
    args = parser.parse_args()
    return args