4
1
test
发布于

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

著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

浏览 (116)
点赞 (4)
收藏
评论 1
云端探险者-1820
云端探险者-1820
323
点赞
评论