Update readme

README.md

@@ -10,7 +10,7 @@ library_name: diffusers
 
 ![Intro Image](sdxl_lightning_samples.jpg)
 
-SDXL-Lightning is a lightning-fast text-to-image generation model. It can generate high-quality 1024px images in a few steps. For more information, please refer to our paper: [SDXL-Lightning: Progressive Adversarial Diffusion Distillation](https://huggingface.co/ByteDance/SDXL-Lightning/resolve/main/sdxl_lightning_report.pdf). The models are released for research purposes only.
+SDXL-Lightning is a lightning-fast text-to-image generation model. It can generate high-quality 1024px images in a few steps. For more information, please refer to our paper: [SDXL-Lightning: Progressive Adversarial Diffusion Distillation](https://arxiv.org/abs/2402.13929). The models are released for research purposes only.
 
 Our models are distilled from [stabilityai/stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0). This repository contains checkpoints for 1-step, 2-step, 4-step, and 8-step distilled models. The generation quality of our 2-step, 4-step, and 8-step model is amazing. Our 1-step model is more experimental.
 
@@ -125,4 +125,17 @@ The 1-step model is only experimental and the quality is much less stable. Consi
 1. Download the full checkpoint (`sdxl_lightning_1step_x0.safetensors`) to `/ComfyUI/models/checkpoints`.
 1. Download our [ComfyUI full 1-step workflow](comfyui/sdxl_lightning_workflow_full_1step.json).
 
-![SDXL-Lightning ComfyUI Full 1-Step Workflow](comfyui/sdxl_lightning_workflow_full_1step.jpg)
+![SDXL-Lightning ComfyUI Full 1-Step Workflow](comfyui/sdxl_lightning_workflow_full_1step.jpg)
+
+
+## Cite Our Work
+```
+@misc{lin2024sdxllightning,
+      title={SDXL-Lightning: Progressive Adversarial Diffusion Distillation}, 
+      author={Shanchuan Lin and Anran Wang and Xiao Yang},
+      year={2024},
+      eprint={2402.13929},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+}
+```