Function main

demo/HuggingFace/run.py:282–308

Parses network folders and responsible for passing --help flags to subcommands if --network is provided.

()

def main() -> None:
    """
    Parses network folders and responsible for passing --help flags to subcommands if --network is provided.
    """
    # Verify python version support
    verify_python_version()

    # Get all available network scripts
    networks = register_network_folders(os.getcwd())

    # Add network folder for entry point
    description = "Runs TensorRT networks that are based-off of HuggingFace variants."
    parser = get_default_parser(networks, description, add_default_help=False)

    # Get the general network wrapper help
    known_args, _ = parser.parse_known_args()

    # Delegate parser to action specifics
    action = get_action(known_args.action, networks, parser)
    known_args, _ = parser.parse_known_args()

    # If bootstrap occurs, then the spawned process completes the rest of demo.
    # We can exit safely. We spawn after parsing basic args to reduce loading churn on rudimentary help commands.
    if bootstrap_ld_library_path():
        sys.exit(0)

    return action.execute(known_args)


if __name__ == "__main__":