参考:【成本1.3元】定制你的AI画画大模型,dreambooth 训练风格/人物_哔哩哔哩_bilibili
一、创建实例
二、更新环境(也许不用做)
更新accelerate, transformer, diffusers
三、下载sd1-5
sd1-5 https://www.alipan.com/s/ANKHixJcowW 提取码: 48ib 点击链接保存,或者复制本段内容,打开「阿里云盘」APP ,无需下载极速在线查看,视频原画倍速播放。
四、修改train_object.sh
删去PPL相关行(4行)
更改模型路径、数据集路径等
五、修改train_dreambooth.py
将main(刚400行左右)的accelerator = Accelerator(
...
logging_dir=logging_dir,
)
改为project_dir=logging_dir
其余和视频保持一致