Unofficial PyTorch implementation of "Mean Flows for One-step Generative Modeling" (2025). Extends the framework to Edge-to-Image generation with Multi-scale Edge Encoder and 1-NFE sampling.
computer-vision deep-learning pytorch generative-model image-generation diffusion-models flow-matching meanflow one-step-generation
-
Updated
Jan 14, 2026 - Python