TAO技能库能力查询Skill tao-list-capabilities

该技能用于回答TAO技能库(TAO Skill Bank)插件能够做什么,包括列出可用的应用程序、数据工作流、模型列表、平台和AutoML支持。通过运行辅助脚本生成能力清单,用于用户询问TAO技能库功能、TAO模型、工作流和AutoML支持时提供答案。关键词:TAO技能库, 能力列表, 模型清单, AutoML支持, 工作流发现, NVIDIA TAO。

技能发现 0 次安装 1 次浏览 更新于 9/6/2026
名称 tao-list-capabilities
描述 >- 通过从打包的应用程序、数据、模型、AutoML和平台清单中生成响应,回答TAO技能库插件能做什么。当用户询问"TAO技能库能做什么"、“列出TAO模型”、"哪些TAO工作流可用"或"什么支持AutoML"时使用。
开源协议 Apache-2.0 compatibility: 需要打包的TAO技能库辅助脚本。 metadata:
作者 NVIDIA Corporation
版本 “0.1.0” allowed-tools: 读取Bash tags: - tao - capabilities - discovery

TAO技能库能力

独立安装? 如果此会话不是由TAO技能库插件初始化的,请先运行tao-setup技能(主机预检、凭证、跨技能发现)。

当用户询问tao-skill-bank能做什么、询问插件能力、询问哪些应用程序或数据工作流可用、询问支持哪些模型,或询问哪些模型支持AutoML时,使用此技能。

快速开始

对于一般能力问题,运行scripts/list_tao_capabilities.py;对于模型/动作和AutoML支持问题,运行scripts/list_tao_models.py

能力答案

对于一般能力答案,运行打包的辅助脚本:

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_capabilities.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --format text

使用辅助输出作为答案的来源,而不是手动枚举此技能或插件元数据的能力。包括:

  • applications/ 下的每个顶级应用程序工作流及其功能。
  • data/ 下的每个顶级数据工作流及其功能。
  • 来自 scripts/list_tao_platforms.py 的支持执行平台。
  • models/ 下模型的微调/部署工作流覆盖范围:当打包的模式清单中存在这些动作时,包括训练、评估、推理、导出和TensorRT引擎生成。
  • AutoML支持以及AutoML每个动作的模式门控。

模型列表

当用户询问哪些TAO模型可用或模型可以运行哪些动作时,使用打包的模型列表脚本,而不是手动扫描模型文件夹:

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_models.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --scope all --format text

模型列表来自skills/models/schemas.manifest.json

AutoML列表

当用户询问哪些模型支持AutoML时,使用相同的模型列表脚本在AutoML模式下,或兼容性包装器:

${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_tao_models.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --scope automl --format text
${TAO_SKILL_BANK_PATH:-~/tao-skills-external}/scripts/list_automl_support.py \
  --skill-bank ${TAO_SKILL_BANK_PATH:-~/tao-skills-external} --format text

当用户询问特定压缩动作时,使用--action distill--action prune--action quantize

AutoML对动作的支持要求skills/models/<model_skill>/schemas/<action>.schema.json与插件一起打包,并能成功解析为JSON。如果该数据类架构缺失或无效,不要将模型/动作描述为AutoML支持。