当前位置:   article > 正文

SD-Trainer安装及使用-Dreambooth_sd trainer

sd trainer

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开始启动,启动后的页面与秋叶版一致。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/运维做开发/article/detail/895117
推荐阅读
相关标签
  

闽ICP备14008679号