使用Lora-script训练Lora模型

626 阅读1分钟

下载Lora-script

克隆带子模块的仓库,带上子模块,不然在使用的过程中会有错误

git clone --recurse-submodules https://github.com/Akegarasu/lora-scripts

创建环境

也可以使用自己本地python环境

conda create -n lora python=3.10.11

安装依赖

Windows
安装
运行 install-cn.ps1 将自动为您创建虚拟环境并安装必要的依赖。

训练
运行 run_gui.ps1,程序将自动打开 http://127.0.0.1:28000


Linux
安装
运行 install.bash 将创建虚拟环境并安装必要的依赖。

训练
运行 run_gui.sh,程序将自动打开 http://127.0.0.1:28000

图片打标

选择WD标签器

  • 设置图片目录

点击启动

查看进度

第一次启动会先下载模型

在控制台查看打标进度

打标完成后在图片目录下就可以看到对应的txt文件

标签调整

如果模型打标的结果不是很满意的话,可以自己手动调整

  • 选择标签编辑器
  • 填写数据目录
  • 点击加载

通过勾选不同的标签,过滤出需要调整的图片

删除标签

最后点击 Save all changes保存

训练Lora

  • 选择LoRA训练
  • 选择底模路径
  • 选择刚才打标的图片和标签所在的目录
  • 修改模型名称
  • 输出模型的路径
  • 点击开始训练

训练启动

进度查看

训练完成后查看模型,找到刚才配置的模型输出目录

使用LoRA模型

在Stable Diffusion中,选择Lora,点击对应的Lora模型。点击生成