Stable Diffusion Lora模型训练详细教程

1. 介绍

通过Lora小模型可以控制很多特定场景的内容生成。
但是那些模型是别人训练好的，你肯定很好奇，我也想训练一个自己的专属模型（也叫炼丹～_～）。
甚至可以训练一个专属家庭版的模型（family model），非常有意思。
将自己的训练好的Lora模型放到stableDiffusion lora 目录中，同时配上美丽的封面图。


2. 模型训练步骤

2.1 训练环境搭建

• WebUI或者Diffuser
  https://github.com/AUTOMATIC1111/stable-diffusion-webui
  
• Lora训练环境
  https://github.com/kohya-ss/sd-scripts
  

2.2 数据准备

从网上爬取一些想要的角色图片，或者直接去截图。
这次我做的初春，找一些图片就行了，不需要很多，20张就可以，各个角度，全身，大头尽量都有些。
角色是这样子的，头上一定得有花，这个是角色属性。

这些是我找的图：

2.3 数据清洗打标

得到数据后，第一步就是清洗，所谓清洗主要是把爬的垃圾数据删了，并且该抠图的抠图（角色的Lora其实不扣也不太影响，真人是尽力抠图，功能性的lora基本上是要手动p图的）。
这里我给大家一个抠图脚本，用的阿里云API，非常方便：

1. 链接：https://pan.baidu.com/s/1PdF2ocgqOBtRmQqtmij6RA?pwd=bjf4
2. 提取码：bjf4

复制代码

总之清晰后数据就是干净的，最好是扣过的图，只有人物主体。
然后我们就要打标了，如果是真人或者风格类lora，可以学youtube里那些人做法，直接用BLIP做image caption，然后手动修改一些。但是二次元强烈建议直接上deepbooru，这是因为二次元SD的base model源头是NovelAI泄露的，而当时的模型就是这个风格的标签，所以二次元特别适合用booru风格的描述。具体操作如下：
首先打开SD webui，找到如下地方：

至于说分辨率，其实512就可以了，可以调大一些，如果你显存够大，我用3090发现基本上拉到1024分辨率后就没有收益了，而低于512明显效果不好。理论上图片分辨率高一些好，此外图片质量解析力也应该高。
打完勾后，就可以process了。

当程序自动走完流程我们就在目标文件夹里得到一组由image和text文件组成的对：

其中txt文件里就是对一幅图的描述，如下：

1. 1girl, blurry, solo, depth_of_field, hairband, blurry_background, building, torii

复制代码

然后就进行最重要的步骤，打标。
所谓打标就是监督学习，告诉SD我们要它学什么。这里我们就是想学一个角色，这个角色有很多特征，比如她头上有花花，短发，眼睛样子等。
因此我们要把这些角色特征的描述词，从txt文件中的描述中删除。
。。。。。。
然后我们再加入一个特定的角色名，用来表示这个角色。让这个角色名学到刚才我们删除的特征。
换句话说，如果我们不删除那些特征，模型是不会把这些特征学给这个角色名的，而是专门学给具体的描述词。比如如果我们留下来头上的花，那么头上有花这个概念就不会学给初春这个角色名。当我们生成图像时候，初春这个词是不会生成头上有花的女孩，但是初春必须头上有花！所以我们要干掉这个词，这样模型就认为初春头上必然有花。
其实就很简单，控制变量法，我们留下来的标签就是我们想让模型学习的。
但是编辑这玩意如果一个一个文件走会很烦，我这里有个简单的步骤：
首先下载一个webui的插件“
stable-diffusion-webui-dataset-tag-editor-main
（请去Github搜，然后放进Webui的Extensions目录下），然后我们可以进行批量编辑了：


这就是可以批量编辑的地方，我们进行批量remove，即批量删除我们不想要的特征词：

这些词我认为都是初春特征，所以我都打勾删除，记得删除后保存下。
等到这些删除完了后，我们在每句描述前面都加上uiharu_kazari，这个是初春的触发词。一旦我们在prompts写上这个词，初春就会出现！我认为很重要，当然有人说可以不加，其实我训练几十个lora很多方案都试过，现在我感觉也是可加可不加的，但是我喜欢加上，里面原因很难直接说，就是加上后出图效果会好一些。但是不一定，有时候会变差，我说的这个方案其实是有些晋级的了，新人建议全都不删除tags，标签出来直接都拿去train，最简单稳定。
为什么放在第一位，实际上tags是有顺序的，一般最开始的tags的权重最大，越靠后越小，所以我们要把我们最想学的这个人放在第一位，像是没有意义的概念，比如透明背景我们尽量往后放。
此外我们需要看每张图的描述，有一些描述是含有明显错误的，因为毕竟是打标器，很多都是错的。我们一定要删除错误的标签，这些错误是会明显影响我们训练结果的（当然图多时候其实不影响了，但是话又说回来，图片质量>图片数量，风格、功能性Lora>真人>二次元，二次元是最简单的lora）。
2.4 训练

我们开始写一个训练脚本参数。
首先是Base Model概念，这个应该是最重要概念之一，就是我们是在什么模型基础上训练的。
比如我们训练真人，女性，社区目前首选Chilloutmix系列。如果是汽车之类的物体，首选可以是SD1.5 2.1等官方模型。如果是二次元，我们这次首选NovelAI系列的模型，我这里选的Acertain，因为它画风比较朴素，接近动漫效果，我感觉用来训练比较好。
我这次直接把脚本参数都列出来了：

1. # LoRA train script by @Akegarasu
2. # Train data path | 设置训练用模型、图片
3. $pretrained_model="D:\workspace\stable-diffusion-webui\models\Stable-diffusion\ACertainModel.ckpt" # base model path | 底模路径
4. $train_data_dir="C:\Users\NUOSEN\Desktop\data\lora\Kazari
5. "
6. $reg_data_dir = "" # directory for regularization images | 正则化数据集路径，默认不使用正则化图像。
8. # Train related params | 训练相关参数
9. $resolution = "768,768" # image resolution w,h. 图片分辨率，宽,高。支持非正方形，但必须是 64 倍数。
10. $batch_size = 2 # batch size
11. $max_train_epoches = 10 # max train epoches | 最大训练 epoch
12. $save_every_n_epochs = 1 # save every n epochs | 每 N 个 epoch 保存一次
13. $network_dim = 64 # network dim | 常用 4~128，不是越大越好
14. $network_alpha = 32 # network alpha | 常用与 network_dim 相同的值或者采用较小的值，如 network_dim的一半 防止下溢。默认值为 1，使用较小的 alpha 需要提升学习率。
15. $clip_skip = 2 # clip skip | 玄学 一般用 2
16. $train_unet_only = 0 # train U-Net only | 仅训练 U-Net，开启这个会牺牲效果大幅减少显存使用。6G显存可以开启
17. $train_text_encoder_only = 0 # train Text Encoder only | 仅训练 文本编码器
19. # Learning rate | 学习率
20. $lr = "5e-5"
21. $unet_lr = "5e-5"
22. $text_encoder_lr = "6e-6"
23. $lr_scheduler = "cosine_with_restarts" # "linear", "cosine", "cosine_with_restarts", "polynomial", "constant", "constant_with_warmup"
24. $lr_warmup_steps = 50 # warmup steps | 仅在 lr_scheduler 为 constant_with_warmup 时需要填写这个值
25. $lr_restart_cycles = 1 # cosine_with_restarts restart cycles | 余弦退火重启次数，仅在 lr_scheduler 为 cosine_with_restarts 时起效。
27. # Output settings | 输出设置
28. $output_name = "Kazari_v1" # output model name | 模型保存名称
29. $save_model_as = "safetensors" # model save ext | 模型保存格式 ckpt, pt, safetensors
31. # 其他设置
32. $network_weights = "" # pretrained weights for LoRA network | 若需要从已有的 LoRA 模型上继续训练，请填写 LoRA 模型路径。
33. # $network_weights = "D:\workspace\stable-diffusion-webui\models\Lora\koreanDollLikeness_v10.safetensors" # pretrained weights for LoRA network | 若需要从已有的 LoRA 模型上继续训练，请填写 LoRA 模型路径。
34. $min_bucket_reso = 256 # arb min resolution | arb 最小分辨率
35. $max_bucket_reso = 1024 # arb max resolution | arb 最大分辨率
36. $persistent_data_loader_workers = 0 # persistent dataloader workers | 容易爆内存，保留加载训练集的worker，减少每个 epoch 之间的停顿
38. # 优化器设置
39. $use_8bit_adam = 0 # use 8bit adam optimizer | 使用 8bit adam 优化器节省显存，默认启用。部分 10 系老显卡无法使用，修改为 0 禁用。
40. $use_lion = 1 # use lion optimizer | 使用 Lion 优化器
43. # ============= DO NOT MODIFY CONTENTS BELOW | 请勿修改下方内容 =====================
44. # Activate python venv
45. .\venv\Scripts\activate
47. $Env:HF_HOME = "huggingface"
48. $ext_args = [System.Collections.ArrayList]::new()
50. if ($train_unet_only) {
51.   [void]$ext_args.Add("--network_train_unet_only")
52. }
54. if ($train_text_encoder_only) {
55.   [void]$ext_args.Add("--network_train_text_encoder_only")
56. }
58. if ($network_weights) {
59.   [void]$ext_args.Add("--network_weights=" + $network_weights)
60. }
62. if ($reg_data_dir) {
63.   [void]$ext_args.Add("--reg_data_dir=" + $reg_data_dir)
64. }
66. if ($use_8bit_adam) {
67.   [void]$ext_args.Add("--use_8bit_adam")
68. }
70. if ($use_lion) {
71.   [void]$ext_args.Add("--use_lion_optimizer")
72. }
74. if ($persistent_data_loader_workers) {
75.   [void]$ext_args.Add("--persistent_data_loader_workers")
76. }
78. # run train
79. accelerate launch --num_cpu_threads_per_process=8 "./sd-scripts/train_network.py" `
80.   --enable_bucket `
81.   --pretrained_model_name_or_path=$pretrained_model `
82.   --train_data_dir=$train_data_dir `
83.   --output_dir="./output" `
84.   --logging_dir="./logs" `
85.   --resolution=$resolution `
86.   --network_module=networks.lora `
87.   --max_train_epochs=$max_train_epoches `
88.   --learning_rate=$lr `
89.   --unet_lr=$unet_lr `
90.   --text_encoder_lr=$text_encoder_lr `
91.   --lr_scheduler=$lr_scheduler `
92.   --lr_warmup_steps=$lr_warmup_steps `
93.   --lr_scheduler_num_cycles=$lr_restart_cycles `
94.   --network_dim=$network_dim `
95.   --network_alpha=$network_alpha `
96.   --output_name=$output_name `
97.   --train_batch_size=$batch_size `
98.   --save_every_n_epochs=$save_every_n_epochs `
99.   --mixed_precision="fp16" `
100.   --save_precision="fp16" `
101.   --seed="1337" `
102.   --cache_latents `
103.   --clip_skip=$clip_skip `
104.   --prior_loss_weight=1 `
105.   --max_token_length=225 `
106.   --caption_extension=".txt" `
107.   --save_model_as=$save_model_as `
108.   --min_bucket_reso=$min_bucket_reso `
109.   --max_bucket_reso=$max_bucket_reso `
110.   --xformers --shuffle_caption $ext_args
112. Write-Output "Train finished"
113. Read-Host | Out-Null ;

复制代码

train_data_dir是我们放图片的目录，我们把刚才处理好的图片文件夹（里面有图片和文字对）放在这个目录下。
我的目录结构是这样的（其中2_images就是我们处理好的图片文件夹，为什么这么命名一会讲）：

1. C:\Users\NUOSEN\Desktop\data\lora\Kazari
2. \2_uiharu_kazari

复制代码

所以我在上面的脚本里填写的目录为

1. C:\Users\NUOSEN\Desktop\data\lora\Kazari

复制代码

训练的时候脚本自动在Kazari目录下找图片文件夹，它判定的是[NUM]_XXX命名的文件夹就是要输入给dataloader的图片文件夹。
我们一定要按照这个规则明明我们的文件夹，NUM表示我们一个epoch要遍历几遍这个目录，即repeats，而XXX就是我们默认的触发词。
而我们还会在脚本设置max_train_epoches这个参数，这个就表示我们要训练多少个epoch。
所以训练时候优化的次数Step就是num_images X repeats X max_train_epoches / batch_size
batch_size就是一次处理几张图片，也是我们在脚本里设置的，显卡有能力就别设置1。
更重要的参数就是学习率Lr了：
$lr=“5e-5”
$unet_lr=“5e-5”
$text_encoder_lr=“6e-6”
这个我们可以先设置成这样，然后看loss函数下降了没，再进行调整。
训练图片分辨率resolution这个参数也是非常重要的，它是我们训练时候图片的分辨率，显卡有能力设置大一些比较好，建议不要低于512，会很差。
network_dim表示我们训练出来的Lora模型大小，一般不要大于128，因为没收益，小一些的dim可以抗过拟合。
clip_skip这个参数在二次元模型里就设置2或者3，但真人模型可以考虑设置1，但是需要测试效果，一般就2。这个参数其实是CLIP这个模型结构的原因，它其实是层级概念，比如上一层是“人”，那么下一层就是“男人”“女人”，逐渐细化。而二次元这个从NovelAI起源的模型，都沿用了2这个设定。我们也设置成2。
use_lion这个我设置成1，是因为我想启用Lion这个优化器，因为测试效果发现这个优化器泛化性好一些。
save_every_n_epochs这个我设置成1，就是每个epoch都保存模型，这样最后我有10个模型了。
设置好参数我们就一键运行脚本开始训练。

训练完毕。
2.5 检测结果

训练完的lora应该看下不同epoch的结果，排除欠拟合和过拟合的模型，用最合适的模型。
我们直接用简单的prompts遍历所有的模型和weights：

能出一张xy表：

我最后选了8epoch的模型，因为我感觉它在各种环境下最稳定。
2.6 生成图片

训练好的Lora就可以出图啦。


来源：https://blog.csdn.net/weixin_45250844/article/details/130302817
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

In reply to the person asking about tembak ikan online uang asli, agen togel play, cara main slot online, pulsa slot, slot agen toto, bandar slot online terbesar, slot agen, situs pragmatic asli, aplikasi judi online uang asli, situs sbo terbaru,  I highly recommend this conversational tone about tajir4d details or situs asli slot, situs microgaming, live casino online indonesia, situs slot yg resmi, daftar situs slot terbesar di indonesia, permainan pragmatic slot, game slot joker gaming, slot game slot, bermain judi online, play play slot, alongside all this recommended login tajir4d forum which is worth considering with situs judi online terpercaya, slot play online, apa judi slot, situs akun togel terpercaya, joker gaming slot online, anda slot, permainan online slot, slot terpercaya terbaru, main casino slot, idn live game, and don't forget this more tips here about tajir4d link alternatif site which is also great. Also, have a look at this i loved this for login tajir4d info as well as situs singapore slot, live slot online, situs habanero, judi slot itu apa, tembak ikan uang asli, promo judi online, games slot, bermain slot adalah, situs toto slot, main casino online, and don't forget this cool login tajir4d advice and don't forget slot online gampang menang, aplikasi slot online, aplikasi slot online terpercaya, agen judi slot, slot pulsa telkomsel,  more about the author for and don't forget togel 24 jam, game slot asli, web judi togel, spadegaming slot, slot online terbesar,  for good measure. Check more @ Awesome Tajir4D Link Alternatif Website d43_bc8

In reply to the man inquiring about facial contouring with dermal fillers, best treatment for skin tightening, vampire breast lift, best face laser treatment, facial dermal fillers, facial to rejuvenate skin, spa rejuvenate, skin rejuvenation, best face tightening, skin facial treatments,  I highly recommend this top rated laser facial skin treatment advice or best face fillers for wrinkles, prx skin treatment, co2 face lift, non invasive thread lift, cool facial, chemical peel treatment, perfect derma peel, skin care red light therapy, best treatment cellulite, lip filler florida, as well as this additional info for pigmentation treatment Sarasota tips alongside all advatx laser treatment, all med spa, matrix facial, skin care treatment, rejuvenation skin care, spa rejuvenate, cosmetic facial rejuvenation, laser face rejuvenation, skin shrinking, facial and laser treatment, not to mention this moved here for laser spa Sarasota site which is also great. Also, have a look at this recommended laser facial Sarasota url which is worth considering with face rejuvenation procedures, light facials, perfect derma peel treatment, botox dermal fillers, lip enhancement, rf body skin tightening, plump lip injections, aesthetics injectables, chemical peel face lift, non invasive thread lift, alongside all this top laser treatment skin tightening tips on top of prp treatment vampire facial, fibroblast skin, treatment microneedling, rejuvenate 528 regenerative aesthetics medical spa, rejuvenate makeup,  enquiry for not to mention best skin facial, dermal fillers wrinkles, best lip injections, dermal filler face lift, med spa,  for good measure. Check more @ Cool Pigmentation Treatment Sarasota Guide 40_0df8