我这个人,做什么事都喜欢自己上手试试。最近一直在捣鼓AI绘图,想训练出一些更符合自己口味的模型。用那些主流的傻瓜软件,虽然方便,但总感觉差点意思。于是我就开始到处找,有没有什么更野、更底层的工具能用。
从开始找到安装,折腾了一整晚
前几天,我一个圈内的老哥给我推了个东西,就叫“女巫训练师”。他说这玩意儿不是给外行人用的,是那些专门调教模型的人在偷偷用。我一听,这不就对我的胃口了吗?
我当时就坐到了电脑前,直接在搜索栏里敲进了“女巫训练师 最新下载”。结果,光是找个干净的地址就够我喝一壶的了。前面弹出来的那些,全是挂羊头卖狗肉的网站,点进去就让你扫码关注,要么就是让你下个破客户端。我翻了整整三页,一个能用的连接都没找到,心里那个火大。
后来我学聪明了,把关键词改了。我加了“开源”、“GitHub”这些词,才在那些老外扎堆的技术论坛里,找到了真正的线索。在一个不起眼的帖子下面,我扒拉出了一个网盘地址,上面写着“仅供交流,随时删除”。我当时就感觉,这东西应该就是正主没跑了。
文件包非常大,我点下下载按钮,看着那个进度条,我知道今晚又得熬夜了。等到它慢悠悠地全部下完,已经是凌晨两点了。我赶紧双击解压,里面果然是各种Python脚本和配置文档。
-
第一步:安装环境。文档要求Python 3.10.x,我电脑上是3.11。我没办法,只能把旧的卸载干净,重新装了个3.10的版本。
-
第二步:配置依赖。这个环节最折磨人。它要求一堆依赖库,我照着列表一个个敲命令安装。结果不是这个库版本冲突,就是那个库找不到源。我对着报错信息,在网上疯狂搜索,又是升级又是降级,来来回回折腾了一个多小时。
-
第三步:运行测试。环境搞定后,我赶紧运行了那个启动脚本。屏幕上刷过一堆命令行代码,终于,一个黑乎乎的窗口弹了出来,虽然界面贼丑,但它确确实实跑起来了!
实际操练:这东西果然厉害
界面虽然简陋,但是里面的功能那叫一个扎实。它把模型训练的每个环节都拆得清清楚楚。我导入了我之前准备好的底模,然后上传了我精心挑选的一批图。这软件的参数调整得非常细致,光是学习率和批处理大小,就有好几个地方可以让我自己手动设置。
我摸索着调整了几组参数,然后点了“开始训练”。我赶紧看了看显卡占用,百分之九十几,跑得飞快!比我之前用的那个图形化工具,效率高了至少一半。
等了大概一个多小时,训练完成了第一轮。我赶紧把新模型导出来,在绘图软件里测试了一下效果。
结果一出来,我当时就乐了。
我需要的那个细节,之前怎么都跑不出来的那个感觉,这一下就有了。图像的风格收得特别紧,而且质量非常高。这“女巫训练师”虽然名字听着玄乎,但确实是把模型调教得服服帖帖。
我赶紧把我的操作记录下来,尤其是那些配置环境的坑。找下载地址费劲,安装环境更费劲。我得把我的每一步细节都写清楚,省得以后想换台电脑用,又得重新走一遍弯路。
工具虽然冷门又难装,但回报绝对值这个辛苦钱。我打算接下来几天,继续用它来微调其他模型,看看能不能摸出更多门道来。