stable diffusion训练lora全流程

1.下载lora训练器

下载秋葉大佬的模型训练器

视频链接:【AI绘画】模型训练器发布!简单又不失专业的LoRA模型训练一键包_哔哩哔哩_bilibili

下方的网盘链接:

百度盘:https://pan.baidu.com/s/1TBaoLkdJVjk_gPpqbUzZFw

夸克盘:https://pan.quark.cn/s/b4ccd1f635b6

提取码:p8uy

直接下载最新版本的就可以了。

2.解压lora-scripts训练器

我为了方便,和启动器解压到一起了

3.打开lora-scripts训练器

3.1更新训练器(可跳过该步骤,如果运行时遇到错误,回来更新一下)

打开前,我们可以先更新一下训练器。网络不好的话,可以双击,A强制更新-国内加速.bat,按任意键退出。

3.2启动训练器

如果3.1步骤进行过强制更新,该步骤右击A启动脚本.bat,以管理员身份运行打开。

如果3.1步骤没有更新或者已经进行过更新,可以直接双击A启动脚本.bat打开。

4.训练器简介及操作

新手一般直接选择新手模式,可以很快开始训练自己的模型。

4.1训练用模型

训练用模型表示你训练lora时候用的底模,一般用sd1.5那种最原始的模型,底模我演示直接用已有模型majicmixrealistic。

底模文件路径

我的启动器的模型路径在下方,最后结尾添加上 \模型名称。

例如:
C:\Users\Administrator\Documents\AI\sd-webui-aki-v4.8\sd-webui-aki-v4.8\models\Stable-diffusion\majicmixRealistic_v7.safetensors

4.2数据集设置

数据集的意思是你用来训练的图片,一般准备512*512像素大小的一批图片,如果是竖向图片可以考虑512*768大小。

我们可以使用sd自带的工具、ps或者格式工厂来对图片进行批量处理。用格式工厂简单一点。

格式工厂改图

数据集路径

训练器里有自带的train目录,我们直接在这里面新建文件夹,比如我现在要训练一个四妹的lora,我就可以在这个文件夹下建立一个文件夹起名叫simei,训练数据集路径填写到该目录。

数据集里面肯定需要有图片的,我们需要进入simei文件夹目录下,再新建一个文件夹我这里起名叫6_simeitu,这里需要解释一下文件夹起名格式需要按照数字下划线名称来写,数字代表里面的每张图片训练的次数,6就代表每张图训练6次。

我们将训练用到的图片放在6_simeitu的文件夹下

保存设置

这里的模型名称可以自己起一个名称,最好是字母别带汉字,模型保存的文件夹我这里默认在output里面了。

caption 选项

这里把训练时随机打乱tokens按钮打开

4.3给训练用到的图片打标签

在打标签之前,先保存参数一下

用WD1.4标签器自动生成标签

这里的path路径填写放训练用到的图片的路径

阈值那里默认是0.5,阈值越低描述的标签越细致。可以考虑改到0.35以上,可以改成0.35。

接着右下角点击启动

我们可以在命令提示框里面看到进度。最终在文件夹下看到一个个文本文档,这里面存放的就是所有图片sd自动推出来的标签。

用标签编辑器进行编辑

标签编辑器可以用来对刚才sd自动推出来的标签进行编辑,我们可以批量删除或者增加标签到所有的txt文本中。

填写训练的图片的路径,点击加载后所有的图片就显示在当前页面上了。

我现在想要给全部的文本文档里面添加一个标签,我自己起名叫simei,根据图中步骤操作后,你可以在文本文档中看到,所有的文档最前面都添加了一个simei的标签。

我现在还需要删除一些没用的标签。选中不想要的标签,点击移除选中的标签

再到最上方保存所有更改,就改好了。

5.开始训练

这时候我们已经可以开始训练模型了

点击开始训练后,就开始跑了

可以看到gpu满载运行中

看到训练完成,说明已经训练好了,我们前往output文件夹可以看到保存的模型。

这里说明一下,有这么多模型是因为,前面我们在保存设置这里,轮写的是2,也就是每2轮保存一次模型,训练10轮,下面是轮数,这里根据个人情况来改

6.测试模型

我们把训练好的lora模型,放在启动器的文件夹中

打开启动器,写入关键词,再把常规起手式点上,采样方法改一下,顺便加个面部修复,就能无限产出了。

注意:训练用到的图质量越好,生成的图质量也会越好。

2024年9月3日