Akegarasu/lora-scripts/lora-train
镜像使用
推荐:使用GUI训练
https://www.bilibili.com/read/cv24050162/
旧版训练 LoRA
打开镜像后 应已经处于 lora-scripts 项目目录。无需安装任何依赖,直接打开
train.sh 进行编辑参数,上传图片至 train 文件夹内。请注意图片文件夹的命名方式。准备好参数和数据后可以开始训练。
打开终端,运行 bash train.sh 即可开始训练。
注意
本镜像所使用的 xformers 基于 sm86 架构编译,本身我是在北京区的3090显卡上制作的这个镜像,推荐你也直接使用3090显卡。
也可以尝试使用这些显卡:Tesla GA10x , RTX Ampere – RTX 3080, GA102 – RTX 3090, RTX A6000, RTX A40
其他显卡不做正常运行保障。如果你用其他显卡,请自行重新安装xformers。
训练数据监控
默认输出日志到 logs 文件夹。使用如下指令即可打开 tensorboard 进行训练数据监控。
tensorboard --logdir=./logs
视频教程(本地)
windows 本地训练教程,大同小异,请自行判断
镜像构建
基本环境
框架及版本
python 3.10
CUDA版本
11.8
构建过程
代码Clone
git clone --recurse-submodules https://github.com/Akegarasu/lora-scripts
依赖安装
bash install.sh
作者:akibanzu
链接:https://www.codewithgpu.com/i/Akegarasu/lora-scripts/lora-train
来源:CodeWithGpu
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。