InstructFlow

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

text2img generation example

Rectflow $^1$ full steps

Rectflow $^1$ 5 steps

Rectflow $^2$ 5 steps

Rectflow $^2$ 1 step

Rectflow $^2$ + Distill 1 step

Rectflow $^2$ + Distill + Midpoint 1 step

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)
• Lattent Adversarial Diffusion Distillaiton (1 step)
• Diffusion RLHF
• Bridge Matching
• Train a model for jpeg restoration, debluring, denoising, superres, coloring
• Train a model for inpainting, outpainting