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、エンベディングといった複数のトレーニング手法をサポートします。自動キャプション（BLIP、BLIP2、WD-1.4）やマスク作成（ClipSeg、Rembg）用の組み込みツールにより、データセット作成がスムーズになります。

Who it’s for

このツールは、AI アーティストや開発者向けに設計されており、コーディングの知識がそれほどなくても、画像・動画生成モデルを高度に制御しながらファインチューニングできるようにします。

Highlights

  • Broad Model Support: SDXL、Stable Cascade、PixArt など、幅広いモデルに対応。
  • Flexible Training: フルファインチューニング、LoRA、エンベディングに加え、マスクトレーニングやマルチ解像度トレーニングのオプションも提供。
  • Dataset Tooling: 自動キャプションとマスク生成ツールを統合。
  • Integrated Sampling: トレーニング中に UI から直接モデルをサンプリングし、進捗をモニタリング可能。
  • Training Optimizations: アスペクト比バケティング、EMA（指数移動平均）サポート、ノイズスケジューラのリスケーリング機能を搭載。