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

治障君在制作灌篮高手真人化这期视频的时候发现了揽睿AI这家新的云显卡平台。
在制作视频的过程中,治障君也把自己训练用的kohya_ss和常用模型做成了镜像和数据集,方便其他有需要的观众使用。以下是如何使用治障君发布的镜像来进行你自己的训练:
首先,登录揽睿,注册一个你自己的账号
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上面下载再上传到自己的文件夹里面。



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


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

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


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

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

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