We are introducing Rectified Flow, a new ODE-based generative model, to this repository (in RectifiedFlow branch). The differences between Rectified Flow and the currently used DDPM will result in some API changes. The testing and adaptation may take one or more weeks. Since we are still in the early stage and the code is not well-organized, the APIs and configurations on the branch may change over time without any backward compatibility. This issue is raised mainly to inform those who are testing and researching on that branch with the changes (and possible migration steps).

TODOs

• Initial implementation with temporary configurations
• Testing, verifying and comparing
• Migrate inference APIs and corresponding configurations to continuous acceleration profile: int64 depth to float32 depth, speedup to steps
• Re-organize code of Rectified Flow, and adapt DDPM to continuous acceleration (convert to discrete settings)
• ONNX exporter adaptations and fixes
• More tests to determine proper default configuration
• Documentation, ready to merge