当前位置:AIGC资讯 > AIGC > 正文

SD-Trainer和Stable diffusion一线牵

SD-Trainer和Stable diffusion一线牵

感谢秋叶大神提供的SD-Trainer和Stable diffusion绘世启动器,让一众爱美的小伙伴能以最快的速度入门AI绘画。
    但真正训练LORA和测试的时候,小伙伴们还是要有很多的时间呆在电脑旁边,手动复制训练文件到Stable diffusion的LOAR目录,还需要在Stable diffusion上修改测试提示词。而且训练LORA总不能尽如人意,需要微调的时候又得要重新操作一遍,耽误不少时间。所以我用RPGDreamer写了个软件“一线牵”,专门解决这个问题。
    只需要在“一线牵”上设置多个SD-Trainer训练任务,并且设置好每一个训练任务的Stable diffusion的绘图测试提示词,就可以让电脑自动训练LORA和调用Stable diffusion绘图测试。
现在让我们以两个不同的测试任务为例看“一线牵”怎么使用。
    “一线牵”是一个仅有4M大小的程序,只负责调用SD-Trainer和Stable diffusion中的训练和绘图功能,不包含任何其它功能,所以在使用前需要手动打开SD-Trainer和Stable diffusion。SD-Trainer需要1.70以上版本,Stable diffusion使用秋叶大神的绘世整合版需要在“高级选项”中开启“启用API”,其它版本需要手动在启动命令行中加入--api。
    现在有两个已经处理好的训练任务在桌面的test目录中,一个是训练某星的脸型,另一个是训练某影视人物的造型。


    打开“一线牵”后按SD-Trainer的使用方式,设置好训练参数。为了演示功能,每个训练都分别用AdamW8Bit和Lion优化器做两个不同的训练测试。
因为SD-Trainer训练的最后一个Lora文件没有编号,不方便Stable diffusion的XYZ图表测试,比如模型保存名称为Mikami_Yua,训练6轮,前五个文件名分别为Mikami_Yua-000001到Mikami_Yua-000005,而最后一个文件名却为Mikami_Yua。所以我在“一线牵”加了一个“训练完成后自动将safetensors文件按序号编号”功能,训练任务完成后会将最后一个文件Mikami_Yua命名为Mikami_Yua-000006。
    做好训练设置后点击“加入任务”,完成当前任务的设置。
    四个不同的训练任务都设置完成再进行Stable diffusion绘图设置。


    在“一线牵”的“SD设置”的正向提示中加入XYZ图表的测试提示词,例如Mikami_Yua的训练任务的正向提示词就需要有Mikami_Yua和<lora:Mikami_Yua-0000NUM:STR>,分别代表LORA的唤醒词和加载不同的LORA文件。
    然后打开X/Y/Z图表功能,X轴类型选择“Seed”,Y轴和Z轴类型都选择“Prompt S/R”。X轴值写“Seed”值,选个小伙伴喜欢的数,我就以100000000为例。Y轴值是加载的LORA文件名,写NUM,01,02,03,04,05,06,Z轴值是加载的LORA强度,写STR,0.6,0.7,0.8,0.9,1。最好打开高清修复功能,这样不会因为像素不够导致脸崩。
做好绘图设置后点击“加入任务”,完成当前绘图的设置。因为每两个训练设置用的是一套训练图像和模型保存名称,所以只需要两个绘图设置。如果需要测试LOAR的泛化性和拟合性,可以加入更多的X/Y/Z图表绘图测试。
    最后在“一线牵”的“训练+测试”界面为每个训练任务加入绘图测试任务。第一、二个训练任务使用唤醒词为“Mikami_Yua”的绘图任务,第三、四个训练任务使用唤醒词为“Wonder_Woman”的绘图任务。


    设置完成后点击“开始”,小伙伴们就可以出门享受生活,第二天来打开“任务目录”,每个训练任务生成的LORA文件和测试绘图都等着小伙伴来”宠幸“呢!
因为一线牵在绘图测试的时候会修改Stable diffusion的保存目录,完成后会自动恢复。如果在后续的使用过程中发现Stable diffusion生成的图像不在默认目录,可以在一线牵的主界面中手动点击“设置SD的Txt2Img输出到默认目录”恢复默认设置。

操作视频:SD-Trainer和Stable diffusion一线牵_哔哩哔哩_bilibili

下载地址:https://pan.baidu.com/s/1btjcjaBoL1agVf4fHg2qcw?pwd=4g06

一线牵.zip可直接使用

一线牵工程.zip为RpgDreamer工程,可做自定义修改

更新时间 2024-06-16