Update README.md

README.md

@@ -10,7 +10,7 @@ datasets:
 - Anthropic/hh-rlhf
 ---
 
-[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. 
+[Pythia-410m](https://huggingface.co/EleutherAI/pythia-410m) supervised finetuned using TRLx library with the helpful subset of [Anthropic-hh-rlhf dataset](https://huggingface.co/datasets/Anthropic/hh-rlhf) for 1 epoch. 
 
 Checkpoints are also uploaded. 
 
@@ -18,5 +18,5 @@ Fully reproducible finetuning code is available on [GitHub](https://github.com/l
 
 [wandb log](https://wandb.ai/lauraomahony999/sft-pythia/runs/quq2097z)
 
-See [Pythia-70m](https://huggingface.co/EleutherAI/pythia-410m) for model details [(paper)](https://arxiv.org/abs/2101.00027). 
+See [Pythia-410m](https://huggingface.co/EleutherAI/pythia-410m) for model details [(paper)](https://arxiv.org/abs/2101.00027).