FACM

--- license: apache-2.0 pipeline_tag: unconditional-image-generation tags: - flow-matching - consistency-models - latent-diffusion - imagenet --- # FACM: Flow-Anchored Consistency Models <h3 align="center"> 🔥 FACM outperforms 2×250-step Lightning-DiT on ImageNet 256 with only 2 steps </h3> <h3 align="center"> <span style="font-size: 1.2em; font-weight: bold;"><strong>FID=1.70 (1-step) &nbsp;&nbsp;&nbsp;&nbsp; FID=1.32 (2-step)</strong></span> </h3> 📄 This is the official implementation of the paper: [Flow-Anchored Consistency Models](https://huggingface.co/papers/2507.03738). Code: [https://github.com/ali-vilab/FACM](https://github.com/ali-vilab/FACM) ## Abstract Continuous-time Consistency Models (CMs) promise efficient few-step generation but face significant challenges with training instability. We argue this instability stems from a fundamental conflict: by training a network to learn only a shortcut across a probability flow, the model loses its grasp on the instantaneous velocity field that defines the flow. Our solution is to explicitly anchor the model in the underlying flow during training. We introduce the Flow-Anchored Consistency Model (FACM), a simple but effective training strategy that uses a Flow Matching (FM) task as an anchor for the primary CM shortcut objective. This Flow-Anchoring approach requires no architectural modifications and is broadly compatible with standard model architectures. By distilling a pre-trained LightningDiT model, our method achieves a state-of-the-art FID of 1.32 with two steps (NFE=2) and 1.76 with just one step (NFE=1) on ImageNet 256x256, significantly outperforming previous methods. This provides a general and effective recipe for building high-performance, few-step generative models. Our code and pretrained models: this https URL . ## ImageNet 256 Performance on 8 × A100 GPUs | Model | Steps | FID | IS | Epochs-Pretrain | Epochs-Distill | Download | |:-----:|:-----:|:---:|:--:|:---------------:|:--------------:|:--------:| | FACM | 2-step | 1.32 | 292 | 800 | 100 | [100ep-stg2.pt](https://huggingface.co/Peterande/FACM/blob/main/100ep-stg2.pt) | | FACM | 1-step | 1.76 | 290 | 800 | 250 | [250ep-stg2.pt](https://huggingface.co/Peterande/FACM/blob/main/250ep-stg2.pt) | | FACM | 1-step | 1.70 | 295 | 800 | 400 | [400ep-stg2.pt](https://huggingface.co/Peterande/FACM/blob/main/400ep-stg2.pt) | ##...

Source context: 0 downloads · 3 likes · Pipeline unconditional-image-generation · Repo Peterande/FACM