haidog-yaqub/MeanFlow
An unofficial PyTorch implementation of MeanFlow, a one-step generative modeling technique based on flow-matching and diffusion models.

Velocity · 7d
+3.0
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of the MeanFlow paper for one-step generative modeling, building on diffusion transformers (DiT) and integrating support for classifier-free guidance. It includes training and inference code for generating images on standard benchmarks like MNIST and CIFAR-10, with multi-GPU support via Hugging Face Accelerate.