快速上手
创建智灵训练器实例
我们只需要在新增 GPU 的时候选择智灵训练器模板,挂载好保存数据的存储,就可以快速启动训练器实例了!
模板选择后,选择需要进行训练的显卡规格,比如示例选择的是一张 1GPU 1×24GB 显存 4CPU 32GB 内存 50GB 临时存储 的显卡。
接下来是挂载存储。你可以挂载已经存在的存储,比如 ComfyUI 所使用的存储。这样,训练好的 LoRA 文件可以直接输出到 /root/ComfyUI/models/loras
目录下,省去了复制的麻烦。
当然你也可以挂载一个新的存储,这里为了方便演示,我们创建一个新的存储。
注意:训练 LoRA 一定要挂载存储!!!
新建的存储大小填写完毕后,我们点击右下角 新增 按钮,就可以创建成功了,等待 GPU 启动完成,就会出现训练器打开按钮了。
点击 打开智灵训练器 按钮,即可进入训练器界面。
数据集打标
训练一个 LoRA 自然是需要一些训练用的数据,智灵训练器提供了两种方式对数据进行打标:
- 训练页进行打标;
- AI 数据集页面进行打标;
两种方式并无功能上的区别,只是 AI 数据集页面占据的页面空间更大,可以预览的更多。
上传图片素材
我们可以在输入框中直接输入图片素材的目录路径,也可以点击右侧的文件夹图标来选择对应目录。
如果没有需要的目录,也可以点击顶部的新建文件夹按钮来创建。这里我们创建一个 images
目录。
创建完成后我们选择 images
目录,就可以去右侧点击上传文件进行上传了。
选择好需要训练的素材就可以开始上传了,上传完成后,可以在右侧看到刚刚上传的素材。
打标
打标模型目前提供了两种:
- Joy Caption2
- Florence2
我们推荐使用 Joy Caption2,因为它对图片的描述偏向自然语言,当然这个看个人需求可自行切换。
我们还支持将触发词也输出到打标文件中,默认这个选项是开启的,你可以在 LoRA 触发词 输入框中输入你想要的触发词,打标时就会自动输出到打标文件中了。
准备好用我们点击一键打标就会将数据集目录中的图片进行打标处理,打标时训练器底部会显示整体的进度,方便你了解打标进度,打标完成后,打标文件就会输出到与图片素材相同的目录下。
打标过程需要加载模型,这可能会花一些时间,请耐心等待。
打标完成后会有弹窗提示。
手动打标或二次编辑
如果你想手动打标,可以在预览界面找到需要打标的图片,右键 --> 手动打标 --> 在底部输入框输入内容,点击保存即可。
除了手动对图片打标,我们还可以编辑已经生成的打标文件,操作方式也是右键菜单:编辑
编辑操作也是一样,在底部输入框中输入内容,点击保存即可。
训练 LoRA
我们提供了两种训练模式选择,分别是:
- 新手模式
- 专家模式
新手模式可调整的训练参数较少,适合初学者,专家模式则是针对有一定经验的用户,可以调整更多的训练参数。
填写必填参数
必填参数包括: LoRA 名称、LoRA 触发词、LoRA 保存路径、数据集目录。
以上参数填写完毕就能开始训练了。
开始训练
点击右下角:开始训练 按钮开始训练,训练过程中会实时更新训练进度,训练完成后生成的 LoRA 文件会输出到刚刚设置的指定目录下,你可以在 LoRA 保存路径 中查看。
训练过程中会实时更新训练进度,如下图所示:
训练完成后,会有弹窗提示,点击确定按钮即可。
此时我们可以去保存的路径查看刚刚生成的 LoRA 文件。