diffusion-pipe 是一个专为扩散模型设计的并行训练脚本,支持大规模模型训练。它具备流水线并行功能,能够高效地利用多GPU资源进行预缓存,并且支持全微调,适用于各种复杂的AI模型训练任务。
Diffusion-Pipe-UI 是一个为扩散模型训练提供一站式解决方案的工具。它通过Gradio界面简化了训练流程,无需复杂命令,支持Docker部署,轻松适配Windows和Linux环境。此外,它还提供Tensorboard可视化训练损失,帮助用户直观掌握训练进度。