Lakonik/pi-Flow-ImageNet

pi-Flow: Policy-Based Flow Models

Distilled 1-step and 2-step ImageNet DiTs proposed in the paper:

pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation
Hansheng Chen¹, Kai Zhang², Hao Tan², Leonidas Guibas¹, Gordon Wetzstein¹, Sai Bi^2
1Stanford University, ²Adobe Research
[arXiv] [Code] [pi-Qwen Demo🤗] [pi-FLUX Demo🤗]

Citation

@misc{piflow,
      title={pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation}, 
      author={Hansheng Chen and Kai Zhang and Hao Tan and Leonidas Guibas and Gordon Wetzstein and Sai Bi},
      year={2025},
      eprint={2510.14974},
      archivePrefix={arXiv},
      primaryClass={cs.LG},
      url={https://arxiv.org/abs/2510.14974}, 
}