OneTrainer: 一款用於訓練與微調各種擴散模型的全方位解決方案

它解決了什麼問題

OneTrainer 提供了一個全面且全方位一體的解決方案，用於訓練與微調擴散模型。它簡化了準備數據集、管理訓練運行以及轉換模型格式的過程，消除了在多個不同的工具之間切換的需求。

運作方式

該軟體提供圖形使用者介面 (GUI) 與命令列介面 (CLI) 來管理訓練流程。它支援廣泛的擴散模型（包括 FLUX.1、Stable Diffusion 1.5 到 3.5，以及 Hunyuan Video）和多種訓練方法，例如全量微調 (full fine-tuning)、LoRA 以及 embeddings。它包含內建工具用於自動標註 (via BLIP, BLIP2, and WD-1.4) 與遮罩建立 (via ClipSeg or Rembg) 以簡化數據集準備工作。

對象是誰

它是為 AI 藝術家與開發者設計的，旨在讓他們能以高度的控制力來微調圖像與影片生成模型，而無需具備深厚的編碼知識。

重點特色

• 廣泛的模型支援：相容於廣泛的模型，包括 SDXL、Stable Cascade 與 PixArt。
• 靈活的訓練：支援全量微調、LoRA 與 embeddings，並提供遮罩訓練與多解析度訓練的選項。
• 數據集工具：整合了自動標註與遮罩生成工具。
• 整合式採樣：能夠在訓練期間直接在 UI 中對模型進行採樣，以監控進度。
• 訓練優化：具備長寬比分桶 (aspect ratio bucketing)、EMA (Exponential Moving Average) 支援，以及雜訊排程器重新縮放 (noise scheduler rescaling)。