Easy LoRA Trainer

104
1.5K

详情

下载文件

模型描述

一个简单快速的LoRA训练器设置,在512像素分辨率、批量大小为1的情况下训练5分钟后,我达到了1步/秒的速度,因此速度相当快。

你可以使用 flux1_devFP8Kijai11GB.safetensors 或常规的 flux1-dev.safetensors 模型。

flux1-dev 略快一些,但加载时间稍长,VRAM占用相同,约为16.3GB,将HighVram设为false对此无任何影响。

将“Split Mode”设为“true”时,最多使用24GB VRAM的41%,即约10GB VRAM占用。

仅当你的VRAM少于16GB时才使用此选项;如果你有16GB,应先尝试设为“false”,看是否会触发OOM。在此特定示例中,设为“true”会使训练时间从20分钟延长至110分钟。

我制作了3个自定义节点,你可以在ComfyUI管理器中找到(缺失节点)。

用于在特定的单个层上训练,区分Layer1和Layer2。

添加了验证预览和损失,并将预览和训练参数保存到目标文件夹。

在512分辨率下训练600步,我只花了15分钟。

此模型生成的图像

未找到图像。