• 小米加入AI大模型之战 推出一款名为SDXS的绘画大模型
  • 发布于 1个月前
  • 68 热度
    0 评论
  • 小熊
  • 0 粉丝 30 篇博客
  •   
最近,国内各大知名互联网公司,都在发布各自的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 的高速图像生成、单步训练以及知识蒸馏。还是为用户提供了更高效、更便捷的解决方案。
用户评论