效果

原理

基于开源项目kohya-ss/sd-scripts增加了Gradio UI，精简了训练参数，更适合初级“炼丹宝宝”的炼丹炉！
核心思想还是比较简单的，通过Gradio UI来收集设置的训练参数，并通过python的 subprocess 库来执行命令 accelerate launch --xxx，最终达到训练LORA的效果。

说明

训练数据集的文件夹在kohya-ss/sd-scripts当前的版本中必须这样指定：
这里 image（当然名字所以，但最好是英文，且不要有空格之类的特殊服务）文件夹作为数据集图片的根目录，子目录需要以数字下划线开头，该数字将影响训练的总步数，即训练的 总步数 = 图片数量 * 子目录的数字前缀

代码

安装步骤可以参考仓库中的说明
点击跳转仓库