小米开源一款新的扩散模型SDXS，让 SD 图像生成速度飙升！

最近，国内各大知名互联网公司，都在发布各自的AI大模型研究成果，这其中不泛有阿里、百度、360、腾讯、华为等，相继在持续‘内卷’。

如今小米也‘内卷’起来了，发布了一款扩散模型新项目（AI绘画领域），名为SDXS。这是要比肩SDXL的节奏哇！

不过它做的就是这一块的，据说相较于SDXL生成速度要提升3倍。

项目介绍

SDXS， 是一种新的扩散模型，通过模型微型化和减少采样步骤，大幅降低了模型延迟。

它利用知识蒸馏来简化U-Net和图像解码器架构，并引入了一种创新的单步DM训练技术，使用特征匹配和分数蒸馆。

SDXS有两款模型系列，分为SDXS-512和SDXS-1024。应该是和图形分辨率强相关的。

两种模型可在单个GPU上分别实现约100 FPS和30 FPS的推理速度，比之前模型快30至60倍。

这可是 SDXL 的 3 倍啊！简直快到让人瞠目结舌！

不仅如此，SDXS 的训练方法在图像条件控制方面具有潜在应用，实现了高效的图像到图像翻译。这意味着它可以广泛应用于快速生成高质量图像、图像上色、草图重构等任务。

目前官方只提供了 SDXS-512-0.9 的模型项目Demo，实测速度真的快到起飞！

Demo：https://huggingface.co/spaces/ameerazam08/SDXS-GPU-Demo

总结

SDXS 可以让SD图像生成速度提升到一个新的台阶。不过个人测试了下，还是生成结果对比SD1.5，感觉从图像清晰度和图像质量还是有些差异，也可能是我的提示词不到位，生成了一些‘不规则’的图像。

而且图像很多地方存在模糊，可能在1024模型下有所改观。

不过 SDXS 的高速图像生成、单步训练以及知识蒸馏。还是为用户提供了更高效、更便捷的解决方案。