End of training
Browse files
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
|