hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
https://hpcaitech.github.io/Open-Sora/
Apache License 2.0
21.54k stars 2.06k forks source link

[feat] **Piecewise** Rectified Flow #109

Closed kabachuha closed 4 months ago

kabachuha commented 5 months ago

As you know from SD3's paper they used Rectified Flow to make the training and the sampling process faster. However, in the past month a new modification of Rectified Flow named PiecewiseRectifiedFlow was released

Project page: https://piecewise-rectified-flow.github.io/ Github: https://github.com/magic-research/piecewise-rectified-flow/tree/main

Claims to be faster than the normal Rectified Flow (used in https://github.com/PKU-YuanGroup/Open-Sora-Plan/issues/43)

I believe it will be a huge quality/speed win compared to the vanilla diffusion pipeline that is used at this moment here

zhengzangw commented 4 months ago

Thanks for your suggestion. Next version will have rectified flow.