XueyingJia commited on
Commit
f45a6ba
·
verified ·
1 Parent(s): 641f14f

End of training

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -1,5 +1,6 @@
1
  ---
2
  base_model: EleutherAI/pythia-1b
 
3
  library_name: transformers
4
  model_name: pythia-1b-sft-HH
5
  tags:
@@ -11,7 +12,7 @@ licence: license
11
 
12
  # Model Card for pythia-1b-sft-HH
13
 
14
- This model is a fine-tuned version of [EleutherAI/pythia-1b](https://huggingface.co/EleutherAI/pythia-1b).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
1
  ---
2
  base_model: EleutherAI/pythia-1b
3
+ datasets: Anthropic/hh-rlhf
4
  library_name: transformers
5
  model_name: pythia-1b-sft-HH
6
  tags:
 
12
 
13
  # Model Card for pythia-1b-sft-HH
14
 
15
+ This model is a fine-tuned version of [EleutherAI/pythia-1b](https://huggingface.co/EleutherAI/pythia-1b) on the [Anthropic/hh-rlhf](https://huggingface.co/datasets/Anthropic/hh-rlhf) dataset.
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
  ## Quick start