← all repositories

lucidrains/rectified-flow-pytorch

PyTorch library implementing rectified flow for generative image synthesis via flow matching.

rectified-flow-pytorch
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

Rectified flow is a generative modeling technique based on flow matching that interpolates between noise and data distributions. This library provides a PyTorch implementation including UNet-based models, reflow training procedures, and sampling utilities for generating images. It supports training with the accelerate library and includes implementations of followup research improvements.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.