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 提供一個無程式碼圖形使用者介面(GUI)與框架,用於微調大型語言模型(LLMs)。它消除了執行複雜微調任務所需的程式設計經驗,讓使用者能無需撰寫程式碼,即可試驗超參數與進階訓練技術。

如何運作

此工具允許使用者上傳資料集並建立實驗以微調 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
  • 專案
  • 專案