赞
踩
SD-Trainer是Stable Diffusion进行LoRA训练的webui。有了SD-Trainer,只需要少许图片,每个人都能够方便快捷地训练出属于自己的模型,进行个性化的创作。
DreamBooth是一种训练技术,通过对主题或风格的少数图像进行训练来更新整个扩散模型。它的工作原理是将提示中的一个特殊单词与示例图像相关联。
我们先选用秋叶版的SD-Trainer,根目录下有几个目录关注一下
双击启动后"A启动脚本"。按如下设置
训练使用模型设置:model_train_type:sd-dreambooth
数据集设置:train_data_dir:./train/dog (指定训练使用数据集目录 )
保存设置:output_name: model_dog (指定生成模型名称)
完成设置后,可以看到整体设置情况。
点击“开始训练”
使用我们生成的模型文件model_dog.safetensors.
如果想自己试着安装一下,可以从以下地址下载源码
https://github.com/Akegarasu/lora-scripts
在Windows下运行install.ps1完成安装。
复制一个基础的模型文件例如v1-5-pruned.safetensors到sd-models目录下
在Windows下运行install.ps1完成安装
在Windows下运行rui-gui.ps1开始启动,启动后的页面与秋叶版一致。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。