Update README.md

README.md

@@ -96,4 +96,18 @@ Here we briefly show our codeSearchNet (codeXGLUE) results between different lay
 - (* size and corresponding projection head present in this model)
 
 ## Licence 
-The model is licensed under the BigCode OpenRAIL-M v1 license agreement. You can find the full agreement [here](https://huggingface.co/spaces/bigcode/bigcode-model-license-agreement).
+The model is licensed under the BigCode OpenRAIL-M v1 license agreement. You can find the full agreement [here](https://huggingface.co/spaces/bigcode/bigcode-model-license-agreement).
+
+
+# Citation
+```
+@article{gurioli2025modeltrainallhierarchical,
+      title={One Model to Train them All: Hierarchical Self-Distillation for Enhanced Early Layer Embeddings}, 
+      author={Andrea Gurioli and Federico Pennino and João Monteiro and Maurizio Gabbrielli},
+      year={2025},
+      eprint={2503.03008},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2503.03008}, 
+}
+```