Exploring and Exploiting Stability in Latent Flow Matching

Official checkpoints for our ICML 2026 paper:

Exploring and Exploiting Stability in Latent Flow Matching

Paper: arXiv:2605.08398
Hugging Face paper page: huggingface.co/papers/2605.08398
Code: github.com/briqr/explo-r-it-ing_lfm_stability

Checkpoints

vaes/
celebhq_models/
ffhq_models/
imagenet_models/

Usage

python -m inference.sample_models \
  --ckpt celebhq_precomputed_unconditionalvaeretrained_fm_balanced_cluster_nearest_s2_pruned05_seed0 \
  --results_dir celebhq_models \
  --mode single