效果

原理

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

界面参数说明

基础模型：后缀为.ckpt或者.safetensors的模型，也可以指定huggingface中的模型id（需要可以顺利访问huggingface）

数据集：训练数据集的文件夹在kohya-ss/sd-scripts当前的版本中必须这样指定：
这里 image&#x