hiyouga/LlamaFactory

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

它解决了什么问题

LlamaFactory 简化了对 100 多种不同大语言模型 (LLMs) 进行微调和部署的过程。它通过提供一个统一、用户友好的界面来进行模型训练、评估和服务,消除了复杂编码的需求,即使是没有深厚机器学习专业知识的用户也能轻松上手。

工作原理

该项目提供了一个模块化框架,支持多种训练方法——从标准的监督微调到 DPO 和 PPO 等高级强化学习技术。用户可以通过零代码 Web UI (LLaMA Board) 或命令行界面 (CLI) 与系统交互。它集成了主流的硬件加速工具并支持多种模型架构,使用户能够以极少的配置在本地或云端训练模型。

适用人群

它专为需要针对特定任务(如工具使用、多轮对话或图像理解)定制大语言模型的开发者、研究人员和 AI 从业者设计,无需从零开始构建复杂的训练流水线。

亮点

  • 广泛的模型支持:兼容 100 多种模型,包括 LLaMA, Mistral, Qwen, DeepSeek 和 Gemma。
  • 零代码界面:包含基于 Gradio 的 Web UI,用于可视化模型管理和训练。
  • 高效训练:支持 QLoRA, FlashAttention-2 和 Unsloth 等先进的内存优化技术,可在消费级硬件上训练大模型。
  • 部署就绪:内置支持使用 vLLM 或 SGLang 进行 OpenAI 风格的 API 部署,实现高速推理。
  • 实验跟踪:集成 TensorBoard, Wandb 和 SwanLab 等工具,实时监控训练进度。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目