DiffBIR
DiffBIR copied to clipboard
推理大一点的图片非常慢,请问作者有推出小模型的计划吗
如题,谢谢!
多久
@fusu192 一张要若干分钟
正常
你好~ 推理速度慢的原因在于UNet,而预训练的UNet结构已经固定了,所以暂时没办法减少它的参数量,除非使用蒸馏过后的sd作为生成先验。我之后更新的重点会放到工程优化上,现在打算先参考sd webui里的一些优化技巧来降低显存占用,然后再看看怎么提高推理速度~
你问了一个现在全世界都很难解决的问题...
慢还有一个原因是是分辨率大,attention的时间复杂度上升导致的。不过模型大小和分辨率,哪个比重更高就不清楚了~