太初玄鼎 · 下一代 Anima 扩散模型训练工作台

MonadForge 训练,由你掌控

MonadForge 是 anima_lora 的深度改进分支,将桌面 GUI 重构为 FastAPI + Vue 3 现代化 WebUI,并加入本地训练守护进程、WebSocket 实时监控、蒸馏、合并、预处理、任务队列与多优化器支持。

Windows / Linux 一键安装 本地 daemon 串行 GPU 队列 中英文双语界面
Training Dashboard Live WebSocket
step 128 实时进度
0.0321 Loss
8.7 GB VRAM
Config Editor 可视化 TOML 字段级编辑、预设创建与实时帮助提示。
Task Queue daemon 串行队列,避免多任务抢占显存导致 OOM。
Distill / Merge Modulation Guidance 蒸馏与多检查点融合。
Vue 3 WebUI
FastAPI
TaskService
Daemon
GPU Guard
Training Run
Overview

从数据准备到模型合并,一站式训练工作台。

MonadForge 保留上游核心训练与推理能力,并补强工程体验:WebUI、任务监控、实时日志、预览图、硬件信息、系统管理与国际化。

0 核心 WebUI 页面覆盖完整训练流程
0 GPU 串行队列与 GPU 守卫,避免显存争抢
0MB 移除 PySide6 后依赖体积显著降低
0 内置 English / 简体中文双语界面
Features

高级训练能力,被封装进更现代的体验。

适合需要长期训练、复杂配置、实时反馈和可恢复任务流的开发者与模型训练者。

实时训练监控

通过 WebSocket 流式推送 step、epoch、loss、lr、速度、预览图与日志,训练状态不再靠反复查看终端。

本地训练守护进程

daemon 作为串行作业队列与 GPU 守卫,任务状态落盘,WebUI 或托盘关闭后仍可继续运行。

可视化配置编辑器

支持方法、变体、预设选择,TOML 字段级编辑、实时帮助提示,以及自定义变体和硬件预设。

数据集浏览器

多目录管理、缩略图网格/列表视图、搜索筛选、排序、图片详情与标签编辑,预处理前更可控。

蒸馏与合并

提供 Modulation Guidance 蒸馏配置,并支持多检查点融合为独立 DiT,方便实验与交付。

多优化器与算法扩展

支持 AdamW、bitsandbytes 8-bit、Paged 优化器、Schedule-Free、Adafactor、CAME 等多种优化策略。

Workflow

像操作控制台一样管理训练。

WebUI 负责提交、订阅与展示;训练、预处理、合并等任务交给本地 daemon,职责清晰,运行更稳。

daemon.log · live stream
$ python tasks.py daemon
[daemon] initializing local training queue...
[daemon] GPU guard enabled: single-job mode
[daemon] WebUI sidecar started at http://127.0.0.1:8000

[task] job queued: lora_training_042
[preprocess] resize / VAE cache / TE cache / PE cache
[monitor] WebSocket metrics stream connected
[train] step 128/8000 · loss 0.0321 · lr 1.2e-5
[preview] sample image refreshed
[queue] next job waiting · GPU memory stable
01

配置训练方案

在 Config 页面选择方法、变体、预设,编辑 TOML 字段,创建自定义训练配置。

02

准备数据集

通过 Dataset 与 Preprocess 管理数据路径、缩略图、标签、缓存、resize、VAE/TE/PE cache。

03

提交守护任务

训练任务进入 daemon 串行队列,由 GPU 守卫调度,浏览器关闭后任务继续运行。

04

监控、蒸馏、合并

实时查看损失曲线、学习率、GPU 状态与预览图,也可执行蒸馏或模型合并。

Start Training

准备好启动你的下一次 Anima 扩散模型训练了吗?

克隆仓库,运行安装脚本,启动 WebUI。MonadForge 会把复杂的训练流程组织成更清晰、更可靠、更具反馈感的体验。