← all repositories

haidog-yaqub/MeanFlow

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

1.2k stars Python Image · Video · Audio
MeanFlow
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.

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