欢迎光临散文网 会员登陆 & 注册

LORA炼丹模型训练教程之线上版

2023-03-31 00:16 作者:觉悟之坡  | 我要投稿

(抱歉昨天没更新,这两天在测试真人lora训练教程,昨晚肝到2点多睡,今天又弄了一天,还没弄好,但尽量会在这几天测试成功发出来。

训练真人的话,很多商业场景都用得上,比如衣服,内衣,项链等模特展示图,婚纱照,个人艺术照等场景使用,所以感兴趣的同学们,先点点关注,然后敬请期待吧~ 

友情提示:非授权的训练真人图像商用及传播行为会有侵权风险,请在法律法规许可范围内使用,教程仅供学术研究参考使用)


上次给大家发出来了线下版lora训练模型教程,但是,很多同学过来提问,说各种报错。

因为本身训练模型的参数很多,还需要下载安装运行环境,各种依赖等,不确定性很大。

而且,训练模型对显卡的性能要求很高,但每个人的显卡常常又各不相同。

想了想,换个线上环境也许会好一些,今天,就给你配置好各种安装环境,依赖软件,还给你一个rtx 3090显卡。

这样的话,出现问题的概率会大大下降,成功几率直线上升。

那么,今天就给大家带来线上版的lora训练教程吧~


(注意,本教程仅适合动漫人物训练)

  1. 1.    首先,打开登陆autodl

(1)打开www.autodl.com,按提示登录。

(2)在算力市场找到3090显卡,然后点击租用。

3090非常抢手,如果抢不到的话,也可以尝试这几个显卡:Tesla GA10x , RTX Ampere – RTX 3080, GA102 – RTX 3090, RTX A6000, RTX A40。其他显卡不要选。

  1. 2.    选择配置环境

选择社区镜像,搜索lora找到下图这个Akegarasu-lora-scripts/lora-train镜像。接着点击立即创建。

  1. 3.    打开实例

等待左侧的状态从准备中,到开机中,到运行中。大概1-2分钟。

然后点击右侧的JupyteLab

  1. 4.    启动终端并更新脚本

(1)点击启动页右侧下面的终端,

(3)输入git pull && git submodule update指令并回车,让系统更新脚本至最新。

  1. 5.    准备图片和预处理

(1)提前准备好要训练的图片文件,记得需要不同角度的,正面,侧面,等图像。不能只有正面大头照。

(2)通过sdwebui-train-preprocess images预处理,系统会自动裁切并给图片打上tag

源路径source就是你自己存放这些原图素材的路径,

目标路径destination就是你要输出的路径,我直接在源路径上加个“2”

(pre process image设置如上图)


(预处理结果)

系统会自动将我们的原素材图裁剪成了512x512,并且给了对应的txt ,即tag文件,还把图片都统一成了png格式。

如果你对自动裁剪的结果不是很满意,可以自己用ps在预处理之前一张一张裁剪调整分辨率,费事一些,但精确并可控。

再把不合适的图片和tag text文件删掉,比如出现了两个人,或者图片里面人物太小的,删掉。

  1. 6.    上传图像

(1)在autodl的GPU终端里面,找到train文件夹,在里面新建一个你想要训练人物的名字,比如我叫做meixuegirl,里面再新建一个文件夹,叫做6_meixuegirl

(2)在6_meixuegirl文件夹里面,上传在第五步里面预处理过的图片和txt文件。


  1. 7.    上传模型

因为这个镜像已经包含了一个动漫模型model.ckpt,我们不用上传。

如果需要自己特殊的动漫模型,则需要上传整个大模型并改名为model.ckpt。

  1. 8.    调整训练参数

(1)左侧双击train.sh打开文件。

(2)修改训练图片文件的路径和名称。

(3)Max ePoches 我改成了20,多训练几次,方便后面好挑一个拟合比较好的。

(4)修改保存名字等参数。(这里设置和本地篇其实是一样的。)

(5)第一次训练,没提到的参数不建议动。

如果你训练成功过,可以按照文件后面的注释去试着调整参数,以便获得更好的训练效果。

(6)修改完保存,Ctrl + S或者点击窗口左上角的“文件”-“保存file”。都可以。

  1. 9.    开始训练

(1)打开终端,输入bash train.sh回车,就会开始训练。

(2)等待训练,大概需要十几分钟

(3)训练完成

  1. 10.  找到生成的safetensor文件(lora)并应用

(1)在output文件夹找到训练完成的meixuegirlxxx.safetensor文件

(2)下载所有safetencor文件并放到你本地电脑的stable-diffusion-webui -> models -Lora文件夹下面

  1. 11.  使用并调试模型

在webui的提示词区域,引用lora并加上相关提示词,生成图像。

lora权重默认给的是1,可以在0.6-0.9之间调试,有时候会有更好的效果。


尝试自己训练的多个lora模型,看看哪个模型表现最好最像的留下,其他的删去。

如果对刚刚训练完的所有lora模型都不满意,则回去第7步调整其他参数不断测试优化。

三.答疑

1.我的人物为什么训练得不像

原因其实可能有很多,

(1)你的训练图质量一定要高,不同角度,正面侧面等。

(2)使用的lora模型也要精挑细选,偏前面的可能会欠拟合,偏后面的可能会过拟合。使用lora权重需要0.6-0.9之间调试。

(3)没有正确调参。需要调整训练文本里吗的参数,多次试验。我这个是成功率比较高的默认参数,但是如果你自己有特殊要求,或者其他变数,需要自己多尝试多调整。


本教程已经尽量详细了,如果还有不会的可以私,会努力回复。

好了,那么今天到课先上到这里吧,明天再见,下课!



参考:

【AI绘画】LoRA 在线云端训练教程 AutoDL , 作者秋叶aaaki

<End>

历史教程文集,请查看我的文集。

记得关注我们,更多AI绘图技能知识持续更新中~

LORA炼丹模型训练教程之线上版的评论 (共 条)

分享到微博请遵守国家法律