Update README.md

README.md

@@ -20,6 +20,16 @@ Fully reproducible finetuning code is available on [GitHub](https://github.com/l
 
 See [Pythia-2.8b](https://huggingface.co/EleutherAI/pythia-2.8b) for model details [(paper)](https://arxiv.org/abs/2101.00027). 
 
+See further details of these models in the paper [Attributing Mode Collapse in the Fine-Tuning of Large Language Models](https://openreview.net/pdf?id=3pDMYjpOxk).
+You can cite these models if they are helpful in your work as follows: 
+
+@inproceedings{o2024attributing,
+  title={Attributing Mode Collapse in the Fine-Tuning of Large Language Models},
+  author={O’Mahony, Laura and Grinsztajn, Leo and Schoelkopf, Hailey and Biderman, Stella},
+  booktitle={ICLR 2024, Mathematical and Empirical Understanding of Foundation Models (ME-FoMo) workshop},
+  year={2024}
+}
+
 hf (pretrained=lomahony/pythia-2.8b-helpful-sft), gen_kwargs: (None), limit: None, num_fewshot: 0, batch_size: 16
 |    Tasks     |Version|Filter|n-shot|    Metric     | Value |   |Stderr|
 |--------------|------:|------|-----:|---------------|------:|---|------|