这个ai-toolkit分支的几个体验优化,训练党会喜欢

分享个我最近在用的ai-toolkit分支,作者做了不少显存优化,主线里没有的那种低垂果实都摘了。更实用的是它把一堆原来只能改配置文件的选项搬到了界面上,比如Prodigy优化器、DoRA支持、还有那种不需要数据集的老式slider lora训练。

日常体验上有几处很戳我:训练队列表格的每一行现在会显示数据集的第一张图,一眼就知道这条任务在练什么;队列还能拖拽重新排序;最舒服的是改采样生成设置不用停训练了,以前为了调个采样得中断挺烦。

另外如果你本地网络或网络环境里有多台带显卡的机器,采样生成能甩给另一台ComfyUI并发跑,不占训练机资源,前提是那台能直接访问训练机的文件系统。

我想问下单机内存够大的话,靠模型换入换出跑采样会不会拖慢主训练。

队列显示首图这个细节太贴心了

改采样不用停训练,就冲这点想换

配置文件里那些选项终于搬到界面了

多机甩采样对有矿场的人友好,我一张卡用不上

单机换入换出肯定会占带宽,多少拖一点

DoRA搬到UI省了不少改配置的功夫

蹲个链接