Lumina-Next

SAFE-DiT 框架加速高分辨率扩散 Transformer 图像生成

研究人员开发了 SAFE-DiT，一个旨在加速使用扩散 Transformer 进行高分辨率图像生成的新框架。该方法解决了“掩码诱导调度税”（MIDT），这是一个系统瓶颈，通过将区域计算编码为注意力掩码来减慢推理速度。SAFE-DiT 将掩码消除与空间调度分离，从而实现更快的处理和更少的内存使用，尤其是在非常高的分辨率下。