训练与微调工具
LoRA Forge:用于训练 Anima、SDXL、FLUX 等模型 LoRA 的 Web UI
PineCookie/LoRA-Forge · Web UI for training LoRAs on Anima, Krea 2, MOSS-Image, SDXL, FLUX, SD3.5 and SD1.5. Tuned for speed: per-block torch.compile, flash-attn, fp16 accumulation and a batched ProdigyPlus step (1.5-2x faster adds up!). Built on kohya-ss sd-scripts and musubi-tuner, with its own MOSS-Image trainer.
ELI5:它是什么,解决什么问题
LoRA Forge 是一个通过浏览器表单启动 LoRA 训练的 Web UI,支持 Anima、Krea 2、MOSS-Image、SDXL、FLUX、SD3.5 和 SD1.5,并为 SDXL 与 SD1.5 提供 Dreambooth 全量微调。底层调用 kohya-ss/sd-scripts、musubi-tuner 以及自研的 MOSS-Image 训练器,自动准备 Python、PyTorch 和训练环境。项目提供 Essentials 简化表单、配置导入导出、Runs 页面进度与 loss 曲线、样本图和控制台输出,并通过 torch.compile、flash-attn、fp16 累加和批量化 ProdigyPlus 步骤提升训练速度。
核心能力与快速开始
★ 3 · TypeScript
- 能做什么
- 提供面向 LoRA 训练的本地 Web UI:按模型页面填写表单并开始训练;自动安装 Python、PyTorch、sd-scripts 与 musubi-tuner;支持 Anima、Krea 2、MOSS-Image、SDXL(含 Illustrious、NoobAI、Pony 等)、FLUX、SD3.5、SD1.5,以及 SDXL/SD1.5 的 Dreambooth 全量微调;含配置 JSON/TOML 导入导出、运行记录、loss 与学习率图表、样本图、控制台输出和 LoRA 文件列表;通过 torch.compile、flash-attn、SDPA、fp16 累加、优化器步骤批量化及 block swap 加速训练。
- 适用场景
- 用于在本地 NVIDIA GPU 上为 Anima、Krea 2、MOSS-Image、SDXL、FLUX、SD3.5、SD1.5 训练 LoRA;进行 SDXL 或 SD1.5 的 Dreambooth 全量微调;在 16 GB 显卡上训练 Krea 2 与 MOSS-Image;通过浏览器管理训练配置、监控训练过程和查看样本图;导出与复用训练配置。
- 项目形态
- 可直接使用的工具
- 部署方式
- 本地部署:Windows 或 Linux,需要 NVIDIA GPU;下载 Release zip 后运行 run_gui.ps1 或 run_gui.sh,首次启动自动下载 Python、PyTorch 和训练器;默认访问 http://127.0.0.1:28000。
- 主要依赖
- Windows 或 Linux;NVIDIA RTX 20 系列及以上(计算能力 7.5),驱动 580 或更新;推荐 RTX 30 系列及以上以使用 bf16 与 flash-attn;Anima/SDXL LoRA 默认设置约 9 GB 显存,MOSS-Image 与 Krea 2 面向 16 GB 显卡;需要 Git 与 uv;flash-attn 为可选安装项。
- 使用限制
- 不支持的硬件包括 GTX 10 系列及更旧显卡,以及其他品牌 GPU。flash-attn 默认不安装。项目主要由个人维护,README 提示可能存在粗糙之处。速度和显存数据基于特定 RTX 5070 Ti 环境测量,其他显卡收益不同。许可证为 AGPL-3.0,商用条件未说明;模型与训练成本未说明。
- 许可证
- AGPL-3.0
- 最近推送
- 2026-10-17 23:14
资料检查:2026-10-18 07:00 · 卡片整理:2026-10-18 07:00
卡片由 AI 根据仓库资料整理,未进行安装或运行验证
职业应用
程序员
作为本地 Web UI 工具,用于在浏览器表单中配置并启动 SDXL、FLUX、SD1.5 等模型的 LoRA 训练,查看训练进度、loss 曲线、样本图和控制台输出,便于开发者进行图像模型微调实验。
查看内容依据
LoRA Forge is a web UI for training LoRAs. You pick a model, fill in a form in the browser, press Start and follow the training on a live page
由 AI 根据来源材料整理,未进行安装或运行验证
讨论 0
登录后参与讨论,阅读资讯无需注册。
还没有评论,来分享第一个看法。