Update README.md

README.md

@@ -57,4 +57,17 @@ At **200** decoding steps:
 
 # Inference
 
-Use the script in the repo. [inference.py](https://huggingface.co/Harley-ml/MNIST-IMG-390k/blob/main/inference.py)
+Use the script in the repo. [inference.py](https://huggingface.co/Harley-ml/MNIST-IMG-390k/blob/main/inference.py)
+
+### Related Models
+
+# Citation
+
+```bibtex
+@misc{mnist-img-390k,
+  title     = {MNIST-IMG-390k: a Tiny Diffusion Model for Generating Handwritten Digits},
+  author    = {Paul Courneya; Harley-ml},
+  year      = {2026},
+  url       = {https://huggingface.co/Harley-ml/MNIST-IMG-390k}
+}
+```