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

使用云来做Dreambooth, Lora模型训练详细教程

2023-04-13 08:50 作者:人工治障  | 我要投稿

治障君在制作灌篮高手真人化这期视频的时候发现了揽睿AI这家新的云显卡平台。

在制作视频的过程中,治障君也把自己训练用的kohya_ss和常用模型做成了镜像和数据集,方便其他有需要的观众使用。以下是如何使用治障君发布的镜像来进行你自己的训练:


  1. 首先,登录揽睿,注册一个你自己的账号

    https://www.lanrui-ai.com/register

揽睿现在推广期间,如果你使用治障君提供的邀请码注册 3269,你将获得揽睿的新人免费训练礼包一份。


2. 来到工作空间,然后点击右上角的创建工作空间

3. 选择一台云主机,特价的3090应该是1.9一个小时,如果图片不多,只是跑几个epoch的训练,3090一个小时应该绰绰有余。

4. 工作空间的名字可以取自己容易辨认的,运营环境镜像在公有镜像里面选择,kohya_ss,挂载网盘,数据盘可以选择不使用,数据集选择zzj-sd-models,里面有治障君推荐的目前常用的Stable Diffusion Checkpoint模型,以及Lora和embedding。完成之后点击创建实例。如果你想要找更多的模型,也可以在c站www.civitai.com或者炼丹阁www.liandange.com上面下载再上传到自己的文件夹里面。

选择完的界面应该是类似这个样子,预训练模型可以选sd-webui-model,忘记关机的话可以设置一个自动停止时间

5. 创建完成后,实例会进入启动开机状态,这里要稍等片刻(2-5分钟左右),等待服务器启动。

显示运行中的时候,就可以在操作这里点击进入

6. 进入Jupyter之后,data文件夹是你自己的网盘,你的训练数据可以全部存放在这里,长期有效,在你的账户中不会丢失。imported_datasets是治障君发布的数据集,里面你可以找到各种常用的model,imported_models里面也有一些官方放入的常用模型。点击Terminal开启命令去。

7. 按照一下步骤依次键入来启动kohya_ss:


cd /app/kohya_ss

source venv/bin/activate

accelerate config

看到这句话出现后,Please select a choice using the arrow or number keys, and selecting with enter,接下去会出现以下选项,按照如下依次填写

问题1:选This machine

问题2:选No distributed training

问题3:输入no

问题4:输入no

问题5:输入no

问题6:输入all

问题7:选fp16

选完之后,继续复制命令

python kohya_gui.py --server_port 27777 --listen 0.0.0.0

这句输入完成之后,看到下面这段话就说明kohya就启动成功                                                                                                                                  

Running on local URL:  http://0.0.0.0:27777

To create a public link, set `share=True` in `launch()`.


8.回到刚才的工作空间界面,点击实例名称


9. 进入之后,点击复制调试地址

10. 在游览器中粘贴刚刚的地址,回车就能看到ui的画面


11. 以上基本就是kohya的启动方式,治障君在data下面设置了一个叫training_data的文件夹,里面做的是如下设置,你也可以用来参考:

12. 训练的模型路径可以参考以下:

对应的是你的模型在数据集下面的链接,治障君这里用的是anything,你可以把anything的模型名字换成你自己想要的底模的文件名:

/ark-contexts/imported_datasets/zzj-sd-models/Stable-diffusion/Anything_V3.ckpt

文件夹进入顺序:imported_datasets -> zzj-sd-models -> Stable-diffusion/

13. 如果你刚才和治障君使用的是一样的文件夹格式,那你可以直接输入以下的目录来配置你的训练:

/ark-contexts/data/training_data/imgs

/ark-contexts/data/training_data/model

/ark-contexts/data/training_data/log

名字更改成自己想要的模型训练名字

14. 打标的话,图片需要先上传到img下面的文件夹。如果你刚开始接触训练的话,可以参考治障君的这期视频来了解训练的基本设置。

看Lora怎么样训练模型,AI真是把他的样子学到骨子里了!

上传图片,如果你本地打标的话也可以跳过打标这一步,直接上传你的标签文件

15. 最后,填完你的设置内容,直接点Train model,你就可以开始你的炼丹之旅啦。Dreambooth的训练方法也可以参照这个教程。

16. 你的训练进度可以在Jupter的窗口里面查看

17. 最后,炼完的丹在你training_data/model的目录,直接下载取丹即可。enjoy~


使用云来做Dreambooth, Lora模型训练详细教程的评论 (共 条)

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