Update README.md

README.md

@@ -10,11 +10,14 @@ datasets:
 - Anthropic/hh-rlhf
 ---
 
-[Pythia-70m](https://huggingface.co/EleutherAI/pythia-70m) supervised finetuned with helpful subset of [Anthropic-hh-rlhf dataset](https://huggingface.co/datasets/Anthropic/hh-rlhf) for 1 epoch. 
+[Pythia-70m](https://huggingface.co/EleutherAI/pythia-410m) supervised finetuned with helpful subset of [Anthropic-hh-rlhf dataset](https://huggingface.co/datasets/Anthropic/hh-rlhf) for 1 epoch. 
 
-Checkpoints are also included. 
+Checkpoints are also uploaded. 
+
+Fully reproducible finetuning code is available on [GitHub](https://github.com/lomahony/trlx-pythia/tree/main)
 
 [wandb log](https://wandb.ai/lauraomahony999/sft-pythia/runs/3w7e3zmd)
 
-See [Pythia-70m](https://huggingface.co/EleutherAI/pythia-70m) for model details [(paper)](https://arxiv.org/abs/2101.00027). 
+See [Pythia-70m](https://huggingface.co/EleutherAI/pythia-410m) for model details [(paper)](https://arxiv.org/abs/2101.00027). 
+