| 名称 | 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支持。