Update README.md

README.md

@@ -11,6 +11,7 @@ Our submission to the `strict-small` track of the [BabyLM challenge](https://bab
 Baby Llama is a 58M-parameter model, distilled from an ensemble consisting of LLaMA-360M and GPT2-705M, both trained on the `babylm_10M` dataset.
 
 See the associated paper (arXiv number **TBA**) for a detailed discussion of the training procedure and of the model performance.
+The training code is available at [https://github.com/timinar/BabyLlama](https://github.com/timinar/BabyLlama).
 
 ### Hyperparameters for the tasks that require fine-tuning