h2oai/h2o-llmstudio

H2O LLM Studio - a framework and no-code GUI for fine-tuning LLMs. Documentation: https://docs.h2o.ai/h2o-llmstudio/

解决的问题

H2O LLM Studio 提供了一个用于微调大型语言模型(LLMs)的无代码图形用户界面(GUI)和框架。它消除了执行复杂微调任务所需的编程经验,使用户能够无需编码即可尝试超参数和高级训练技术。

如何工作

该工具允许用户上传数据集并创建实验以微调 LLM。它同时支持直观管理的 GUI 和面向高级用户的命令行界面(CLI)。该框架与 NVIDIA GPU 集成,并支持低秩适应(LoRA)和 8 位训练等内存高效技术。对于更大的模型,它利用 DeepSpeed 在多个 GPU 上进行分片训练。

适用人群

专为希望无需编写代码即可微调 LLM 的用户,以及需要可视化方式跟踪、比较和评估模型性能的开发者设计。

主要亮点

  • 无代码 GUI:用于管理整个微调过程的专用界面。
  • 高级微调:支持 LoRA、8 位训练以及 DPO、IPO 和 KTO 等偏好优化技术。
  • 性能追踪:可视化比较实验,并与 Weights & Biases (W&B) 集成。
  • 模型导出:可轻松将训练好的模型导出至 Hugging Face Hub。
  • 灵活部署:可通过 Docker、CLI 或云实例(例如 RunPod)运行。
  • 多样化的任务类型:支持因果回归和因果分类建模。

相关

  • 项目
  • Dispatch
  • Dispatch
  • 项目
  • 项目