`nemo_automodel.cli.query_capabilities`#

Thin CLI wrapper around :func:nemo_automodel.query_capabilities.

Usage::

python -m nemo_automodel.cli.query_capabilities --model-id meta-llama/Llama-3.1-8B
python -m nemo_automodel.cli.query_capabilities --model-id <id> --json
python -m nemo_automodel.cli.query_capabilities --model-id <id> --trust-remote-code

Module Contents#

Functions#

`_print_table`
`_parse_args`
`main`	CLI entry point. Returns a process exit code (0 on success).

API#

nemo_automodel.cli.query_capabilities._print_table( model_id: str, caps: nemo_automodel._transformers.model_capabilities.ModelCapabilities, ) → None#

nemo_automodel.cli.query_capabilities._parse_args(argv: list[str] | None = None) → argparse.Namespace#

nemo_automodel.cli.query_capabilities.main(argv: list[str] | None = None) → int#: CLI entry point. Returns a process exit code (0 on success).