AI Toolkit 是专注Stable Diffusion模型训练的开源工具包,主要功能包含:1. 训练脚本优化提供高效训练代码,支持各类Stable Diffusion版本适配,包含SD1.5、SDXL等模型架构。2. FLUX.1训练方案内置FLUX.1训练教程与配置文件模板,详细说明参数设置、数据预处理等关键步骤。3. 多平台适配支持本地电脑和云平台训练,适配RunPod、Modal等常见云服务,提供对应部署指南。4. 可视化操作界面集成Gradio交互界面,通过网页即可完成模型选择、参数配置、训练监控等全流程操作。5. 扩展性支持开放自定义脚本接口,允许用户根据需求调整训练策略,兼容LoRA等主流训练技术。开源地址: