Update README.md

README.md

@@ -14,7 +14,7 @@ license: cc-by-nc-nd-4.0
 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 **108M** LoRA distilled version of [SDXL](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0) model that is able to generate images in **4 steps**. The main purpose of this model is to reproduce the main results of the paper.
-See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart).
+See our [live demo](https://huggingface.co/spaces/jasperai/FlashPixart) and [official implementation](https://github.com/gojasper/flash-diffusion).
 
 
 <p align="center">
@@ -60,6 +60,20 @@ The model was trained for 20k iterations on 4 H100 GPUs (representing approximat
 **Metrics on COCO 2014 validation (Table 3)**
   - FID-10k: 21.62 (4 NFE)
   - CLIP Score: 0.327 (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.