SuperAPI AI 工具集

workbench-llamafactory

这是一个 NVIDIA AI Workbench 示例项目,演示使用 LLaMA-Factory 完成端到端模型开发工作流,用于部署 LLaMA-Factory,并涵盖大语言模型的微调、LoRA 适配器融合、训练后量化以及导出为 TensorRT-LLM checkpoint 等环节,同时支持 unsloth。适合在 RTX AI Toolkit 相关教程指导下进行 LLM 微调实践。

访问官网

工具介绍

用途

本项目是 NVIDIA 提供的 AI Workbench 示例项目,用于部署 LLaMA-Factory,并演示一个端到端的模型开发工作流。其核心目标是在 AI Workbench 环境中完成大语言模型的微调及相关处理流程。

主要能力

根据项目资料,工作流包含以下环节:

  • 使用 LLaMA-Factory 微调大语言模型。
  • 将生成的 LoRA 适配器与基础模型融合(Fuse)。
  • 对模型进行训练后量化(Post Training Quantization)。
  • 将模型导出为 TensorRT-LLM checkpoint。
  • 支持 unsloth。

使用方式

项目为 AI Workbench 项目形式,需在 AI Workbench 中部署 LLaMA-Factory。资料建议按照 RTX AI Toolkit 的 LLM 微调教程进行操作。

环境要求:

  • 最低推荐的 AI Workbench 版本:0.50.16
  • Docker Desktop 版本:4.31+

适用场景

适用于需要在 AI Workbench 环境下体验或实践大语言模型微调、LoRA 融合、训练后量化及 TensorRT-LLM 导出流程的开发者和研究者。

限制

资料未提供支持的模型清单、硬件配置要求、具体命令、参数设置、收费或授权信息,这些内容暂未提供。