Lakonik/LakonLab
Official PyTorch implementations of AsymFlow, pi-Flow, and GMFlow—flow matching models for text-to-image generation published at arXiv, ICLR 2026, and ICML 2025.

This repository contains the official implementations of three research papers on generative models using flow matching. AsymFlow and pi-Flow focus on accelerating text-to-image diffusion models through asymmetric flow designs and imitation distillation respectively, producing high-quality images in fewer steps. GMFlow extends flow matching with Gaussian mixture formulations. The implementations are built in PyTorch with support for ComfyUI integration and HuggingFace demos.