今天跟大家唠唠我最近搞的这个“祖菲”项目,一开始我都没听过这玩意儿,还是朋友推荐,说现在挺火,能生成各种图片,让我试试。
行,既然朋友都说,那就搞起来呗!
第一步:环境准备这玩意儿好像是跑在Python上的,那必须得先把Python环境搞我电脑上原来装个旧版本的Python,寻思着干脆卸重装一个最新的。去Python官网下个安装包,一路“下一步”,总算是装好。然后就是安装依赖包,按照网上的教程,用pip安装torch、transformers这些乱七八糟的库,反正就是照着命令敲,也不太懂是干啥的。
第二步:下载祖菲模型据说祖菲有很多模型,我选一个叫“祖菲2.5”的,听起来比较新。去Hugging Face上找半天,下载一个体积巨大的checkpoint文件。这玩意儿是真的大,足足好几个G,下载的时候我电脑风扇狂转,感觉都要起飞。
第三步:写代码,跑起来!接下来就是写代码。我直接在网上找个祖菲的demo代码,复制粘贴到我的编辑器里。然后修改一下代码,指定模型的路径和一些参数,比如生成图片的数量、尺寸啥的。改完之后,信心满满地运行一下,结果直接报错!
What???
仔细一看,原来是torch版本不对,祖菲2.5需要更高版本的torch。没办法,只能卸载原来的torch,重新安装一个指定版本的。装完之后再次运行,这回总算没报错,但是又出现新的问题:显存不够!
我的电脑显卡比较老,只有4G显存,跑这种大型模型还是有点吃力。降低图片生成的数量和尺寸,勉强跑起来。等好久,终于生成几张图片。
第四步:成果展示生成的图片嘛emmmm,怎么说,有点抽象。我输入一段文字描述:“一只穿着宇航服的猫”,结果生成的图片要么是猫不像猫,要么是宇航服不像宇航服,要么就是两者完全不搭边。不过也有些图片看起来还不错,挺有意思的。
- 尝试各种prompt,发现祖菲对中文的理解能力还是有点弱,生成的效果不太稳定。
- 显存是个大问题,想要生成高质量的图片,必须得升级显卡。
- 祖菲的社区很活跃,有很多大佬分享经验和模型,可以多去学习交流。
这回尝试祖菲的过程还是挺折腾的,遇到不少坑,但也学到很多东西。以后有时间再深入研究一下,看看能不能生成更棒的图片!
就这样,今天的分享就到这里,下次再跟大家聊聊我搞的其他好玩的东西!
还没有评论,来说两句吧...