模型有必要这么大吗?

#3
by jkla139 - opened

模型太大了,实在跑不起来啊!😂

既然动漫图片只是Laion-Chinese的一个子集,我认为不需要这么多参数,削减到500多M应该都可以吧,这样训练和推理都更快
毕竟StyleGAN的模型1024分辨率才300多M

Fengshenbang-LM org

模型太大了,实在跑不起来啊!😂

既然动漫图片只是Laion-Chinese的一个子集,我认为不需要这么多参数,削减到500多M应该都可以吧,这样训练和推理都更快
毕竟StyleGAN的模型1024分辨率才300多M

目前基于diffusion model的模型确实都比较大,我们也考虑后续出个参数量小的版本

有没有办法出只有CPU的版本? @wuxiaojun

Fengshenbang-LM org

有没有办法出只有CPU的版本? @wuxiaojun

神经网络的训练和推理还得是GPU快,CPU肯定是比较慢的天生不支持tensor的并行计算

有没有办法出只有CPU的版本? @wuxiaojun

在i5-7200U上跑,半精度,大约45分钟一张图

Fengshenbang-LM org

有没有办法出只有CPU的版本? @wuxiaojun

在i5-7200U上跑,半精度,大约45分钟一张图

如果你只是推理,可以使用我们提供的动漫webui:https://ccnl.fengshenbang-lm.com/edge/anime/

jkla139 changed discussion status to closed

Sign up or log in to comment