INNER CODE UNIT · Python

main

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

def main():
    args = parse_args()
    logging.basicConfig(level=logging.INFO)

    if args.command == DOWNLOAD_COMMAND:
        LOGGER.info("Downloading files from model...")

        stub = args.stub

        if args.save_dir:
            model = Model(stub, download_path=args.save_dir)
        else:
            model = Model(stub)

        model.download()

        print("Download results")
        print("====================")

View source record →