通过之前的文章,想必你也成功启动了你的QQ机器人,也尝试了AI绘画,如果你对自带的模型绘画能力不满意,可以自己替换。
下面我来教你如果替换默认使用的模型
1.选择一个你满意的模型
打开网站 https://replicate.com/explore 这里有很多可以直接使用的模型
但是需要注意的是并不是所有模型都是绘画类型的,还有些图片编辑、图片识别类型的模型,这些你可以自己研究一下。我们这里先只讲一下如何修改绘画模型。
2.查看模型细节
点击模型的名字,可以直接测试模型的效果 在这里你可以测试效果,如果你还满意的话,可以点击上方的api按钮查看接口需要的数据。 在这里就能看到此模型的坐标和版本号,当然你还可以点击上方的Versions看看有没有更新的版本
3.修改代码
打开 stable_diffusion.py,将上一步获取到的坐标和版本号对号入座
4.检查参数
模型不通,参数也有可能会发生变化,所以我们需要自己检查一下 我们把页面往下拉,可以看到当前这个模型需要的参数 这些灰色方块里的就是参数名,后面 string、integer这些是参数类型 我们需要做的就是看看页面上这些参数代码里是否都有 如果代码里已经包含了这些参数,那么我们不需要改动代码,如果代码里没有某个参数,我们需要自己加上,例如: 页面上有个参数 A,我们需要把他加到代码里 当然,这种写法只是把A的值默认写成了1234,不够灵活,所以我们还是参照上方的代码 这样就可以接受从外部传递进来的参数了
5.修改QBot.py
找到sd_img这个方法,把参数A加上 这里,A的值我直接用默认写死一个了,你可以想办法从对话消息中把你需要的值截取出来,这样就可以实现A参数的动态设置了.
到这里,就修改完了,你可以启动代码测试了。