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（指数移动平均）支持与噪声调度器重新缩放等功能。