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工程，可做自定义修改