Nerogar/OneTrainer

OneTrainer is a one-stop solution for all your Diffusion training needs.

What it solves

OneTrainer 提供一個完整、全方位的解決方案，用於訓練與微調擴散模型。它簡化了資料集準備、訓練執行管理與模型格式轉換的流程，免除在多個不同工具之間切換的需求。

How it works

此軟體同時提供圖形使用者介面（GUI）與指令列介面（CLI）來管理訓練管線。支援多種擴散模型（包括 FLUX.1、Stable Diffusion 1.5 至 3.5、以及 Hunyuan Video）以及多種訓練方式，如完整微調、LoRA 與 embeddings。內建自動標註工具（透過 BLIP、BLIP2 與 WD-1.4）與遮罩生成工具（透過 ClipSeg 或 Rembg），讓資料集準備更為順暢。

Who it’s for

此工具適合 AI 藝術家與開發者，讓他們在不需大量程式撰寫的情況下，對影像與影片生成模型進行高度可控的微調。

Highlights

  • Broad Model Support: 相容於廣泛的模型，包括 SDXL、Stable Cascade 與 PixArt。
  • Flexible Training: 支援完整微調、LoRA 與 embeddings，並提供遮罩訓練與多解析度訓練選項。
  • Dataset Tooling: 整合自動標註與遮罩生成工具。
  • Integrated Sampling: 可在 UI 中直接對模型進行抽樣，以即時監控訓練進度。
  • Training Optimizations: 具備長寬比分桶、EMA（指數移動平均）支援與噪聲排程器重新縮放等功能。