Update README.md

README.md

@@ -12,7 +12,8 @@ inference: False
 
 Flash Diffusion is a diffusion distillation method proposed in [Flash Diffusion: Accelerating Any Conditional
 Diffusion Model for Few Steps Image Generation](http://arxiv.org/abs/2406.02347) *by Clément Chadebec, Onur Tasar, Eyal Benaroche, and Benjamin Aubin.*
-This model is a **66.5M** LoRA distilled version of [Pixart-α](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS) model that is able to generate 1024x1024 images in **4 steps**. See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart).
+This model is a **66.5M** LoRA distilled version of [Pixart-α](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS) model that is able to generate 1024x1024 images in **4 steps**.
+See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart) and [official implentation](https://github.com/gojasper/flash-diffusion).
 
 
 <p align="center">
@@ -70,5 +71,19 @@ The model was trained for 40k iterations on 4 H100 GPUs (representing approximat
   - FID-10k: 29.30 (4 NFE)
   - CLIP Score: 0.303 (4 NFE)
 
+## Citation
+If you find this work useful or use it in your research, please consider citing us
+
+```bibtex
+@misc{chadebec2024flash,
+      title={Flash Diffusion: Accelerating Any Conditional Diffusion Model for Few Steps Image Generation}, 
+      author={Clement Chadebec and Onur Tasar and Eyal Benaroche and Benjamin Aubin},
+      year={2024},
+      eprint={2406.02347},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+}
+```
+
 ## License
 This model is released under the the Creative Commons BY-NC license.