Flow Models

The aim of this repository is to test and implement Flow-Matching-based models

Warning

The data as well as dataset can not be shared!

text2img generation example

Rectflow $^1$ full steps

Rectflow $^1$ 5 steps

Rectflow $^1$ 1 step

Rectflow $^2$ 5 steps

Rectflow $^2$ 1 step

Rectflow $^2$ + Distill 1 step

Rectflow $^2$ + Distill + Midpoint 1 step

LADD 4 steps

Shortcut 4 steps

Inversion example

Input Image	FM Inversed	Changed prompt

TODO:

• Train a foundint text2img model
• FM Inversion
• Midpoint, Euler, RK4 solvers (with CFG)
• N-th Reflow (for straightening and fast simulation)
• Rectified Flow Distillation (1 step)
• Immiscible Diffusion
• Lattent Adversarial Diffusion Distillaiton
• Shortcut Model
• Diffusion RLHF
• Bridge Matching