Update README.md

README.md

@@ -22,6 +22,9 @@ This repository provides **FS-DFM checkpoints** from the paper:
 Amin Karimi Monsefi, Nikhil Bhendawade, Manuel R. Ciosici, Dominic Culver, Yizhe Zhang, Irina Belousova (Jan 9, 2026)  
 ArXiv: 2509.20624
 
+[Github Link](https://github.com/apple/ml-fs-dfm/tree/main)
+[Paper Link](https://arxiv.org/abs/2509.20624)
+
 FS-DFM is a **token-space diffusion / flow-matching language model** designed for **fast long-text generation** by explicitly training for a **user-specified step budget** (e.g., 1–8 steps), while preserving a CTMC-based discrete flow formulation.
 
 ## What’s in this repo
@@ -42,6 +45,15 @@ FS-DFM is a **token-space diffusion / flow-matching language model** designed fo
 
 **Formulation:** discrete flow-matching over a **CTMC** on token sequences; sampling uses custom solvers (e.g., `mixture_euler_with_cumulative_scalar`).
 
+
+## Comparison of Methods
+
+| ARM | DFM | FS-DFM (Ours) |
+|-----|-----|---------------|
+| ![ARM](arm.gif) | ![DFM](dfm.gif) | ![FS-DFM](fs_dfm.gif) |
+
+
+
 ---
 
 ## Architecture