DistriFusion是高分辨率扩散模型的分布式并行推理框架,通过分布式并行推理显著加速了扩散模型在多个GPU上生成高分辨率图像的过程。DistriFusion利用patch parallelism将图像分割并分配给不同设备,无需额外训练即可实现高达6倍的推理速度提升,同时保持图像质量。DistriFusion的实现为AI内容创建和并行计算研究提供了新的基准。

DistriFusion – 高分辨率扩散模型的分布式并行推理框架  第1张
(图片来源网络,侵删)
DistriFusion – 高分辨率扩散模型的分布式并行推理框架  第2张
(图片来源网络,侵删)