AI Toolkit:Stable Diffusion 的 AI 脚本工具集
AI Toolkit:各种 AI 脚本,主要用于 Stable Diffusion
ostris/ai-toolkit 是一个 GitHub 仓库, 包含多种 AI 脚本,主要用于 Stable Diffusion 相关任务。该项目使用 MIT 许可证,目前已获得 4.2k star, 470 forks, 59 watching。
主要功能:
- Stable Diffusion 脚本: 提供各种用于 Stable Diffusion 的脚本和工具。
- FLUX.1 模型训练: 支持对 FLUX.1 系列模型进行训练,包括 FLUX.1-dev 和 FLUX.1-schnell。
- LoRA 和 LoKr 训练: 支持 LoRA (Low-Rank Adaptation) 和 LoKr 训练,允许针对特定层进行训练。
- Gradio UI: 提供一个基于 Gradio 的用户界面,方便用户上传图片、标注、训练和发布 LoRA 模型。
- UI界面: 基于Node.js 和 React 搭建的WEB UI界面, 它可以让你方便地启动、停止、监控训练任务,也能更容易地开始新的训练模型。该UI还处于早期beta阶段,可能存在bug 并且UI安全性未知,不建议暴露到公网。
安装步骤:
……