INNER CODE UNIT · Python

parse_args

neuralmagic/sparsezoo · src/sparsezoo/main.py:270

def parse_args():
    parser = argparse.ArgumentParser(
        description="Download objects from the sparse zoo repo"
    )

    subparsers = parser.add_subparsers(dest="command")
    download_parser = subparsers.add_parser(
        DOWNLOAD_COMMAND,
        description="Download a specific model from the repo.",
    )
    search_parser = subparsers.add_parser(
        SEARCH_COMMAND,
        description="Search for objects from the repo.",
    )

    download_parser.add_argument(
        "stub",
        help="Model stub. Please visit sparsezoo.neuralmagic.com to obtain",

View source record →