Add link to paper and code, improve model card structure

README.md

@@ -1,24 +1,37 @@
 ---
-license: apache-2.0
+base_model:
+- FireRedTeam/FireRed-Image-Edit-1.0
 language:
 - en
 - zh
-base_model:
-- FireRedTeam/FireRed-Image-Edit-1.0
+library_name: diffusers
+license: apache-2.0
 pipeline_tag: image-to-image
 tags:
 - FireRed-Image-Edit-1.0
 - distillation
 - LoRA
-library_name: diffusers
 ---
 
-Please refer to [FireRed-Image-Edit-1.0](https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.0) to learn how to use the distilled edit lora.
+# FireRed-Image-Edit-1.0-Lightning
+
+This repository contains the distilled "Lightning" LoRA weights for [FireRed-Image-Edit-1.0](https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.0), enabling fast, high-quality instruction-based image editing in just 8 steps. 
+
+For more technical details, please refer to the [FireRed-Image-Edit-1.0 Technical Report](https://huggingface.co/papers/2602.13344).
 
-use with diffusers 🧨:
+- **Project Page:** [FireRed-Image-Edit-1.0 Space](https://huggingface.co/spaces/FireRedTeam/FireRed-Image-Edit-1.0)
+- **Repository:** [FireRedTeam/FireRed-Image-Edit](https://github.com/FireRedTeam/FireRed-Image-Edit)
+- **Paper:** [FireRed-Image-Edit-1.0 Technical Report](https://huggingface.co/papers/2602.13344)
 
-make sure to install diffusers from `main` (`pip install git+https://github.com/huggingface/diffusers.git`)
+## Use with diffusers 🧨
+
+To use this distilled LoRA, make sure to install `diffusers` from `main`:
+
+```bash
+pip install git+https://github.com/huggingface/diffusers.git
 ```
+
+```python
 from diffusers import QwenImageEditPlusPipeline
 import torch 
 from PIL import Image
@@ -26,8 +39,11 @@ from PIL import Image
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=torch.bfloat16,
 ).to("cuda")
+
+# Load the distilled Lightning LoRA weights
 pipe.load_lora_weights(
-    "FireRedTeam/FireRed-Image-Edit-1.0-Lightning", weight_name="FireRed-Image-Edit-1.0-Lightning-8steps-v1.0.safetensors"
+    "FireRedTeam/FireRed-Image-Edit-1.0-Lightning", 
+    weight_name="FireRed-Image-Edit-1.0-Lightning-8steps-v1.0.safetensors"
 )
 
 prompt = "在书本封面Python的下方，添加一行英文文字2nd Edition"
@@ -41,7 +57,23 @@ image = pipe(
     width = None,
     num_inference_steps = 8,
     generator=torch.manual_seed(0),
-    true_cfg_scale=1.0,  # 不使用标准 CFG
+    true_cfg_scale=1.0,  # Do not use standard CFG for Lightning LoRA
 ).images[0]
+
 image.save("firered_image_edit_fewsteps.png")
+```
+
+## Citation
+
+If you find this work useful, please consider citing:
+
+```bibtex
+@article{firered2026rededit,
+      title={FireRed-Image-Edit-1.0 Technical Report}, 
+      author={Super Intelligence Team},
+      year={2026},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2602.13344}, 
+}
 ```