基于SDXL的更小蒸馏模型，实现秒内生成1K大图

点蓝色字关注“机器学习算法工程师”设为星标，干货直达！Segmind在发布SSD-1B后，又发布了新的蒸馏模型Segmind-Vega，Segmind-Vega模型的UNet参数量只有0.74B，比SD 1.5还小（0.86B），相比SDXL（2.56B）参数量小了70%，速度提升100%。除了发布Segmind-Vega，还同时放出了基于Segmind Vega实现的LCM LoRA：Segmind-VegaRT，只需要2-8步就可以生成高质量图像。Segmind-Vega相比SDXL去掉了很多transformer blocks，最终模型参数量为745M，只有SDXL的30%。Segmind-Vega也是采用模型蒸馏，从三个模型SDXL，ZavyChromaXL和JuggernautXL上蒸馏得到。在A100 80GB显卡下，Segmind-Vega生成一张1024x1024的图像只需要1.39s，而同样配置下SD 1.5生成一张768x768图像需要1.42s，SDXL生成一张1024x1024图像需要3.35s。如果经过编译优化，Segmind-Vega生成一张1024x1024的图像只需要0.86s，真正实现不到1s生成高清大图。下面是Segmind-Vega和SDXL ………………………………