End of training
Browse files- README.md +6 -3
- config.json +1 -1
README.md
CHANGED
@@ -1,17 +1,20 @@
|
|
1 |
---
|
2 |
base_model: gpt2
|
|
|
|
|
3 |
library_name: transformers
|
4 |
-
model_name: gpt2
|
5 |
tags:
|
6 |
- generated_from_trainer
|
|
|
7 |
- trl
|
8 |
- sft
|
9 |
licence: license
|
10 |
---
|
11 |
|
12 |
-
# Model Card for gpt2
|
13 |
|
14 |
-
This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2).
|
15 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
16 |
|
17 |
## Quick start
|
|
|
1 |
---
|
2 |
base_model: gpt2
|
3 |
+
datasets:
|
4 |
+
- yhavinga/mc4_nl_cleaned
|
5 |
library_name: transformers
|
6 |
+
model_name: gpt2
|
7 |
tags:
|
8 |
- generated_from_trainer
|
9 |
+
- alignment-handbook
|
10 |
- trl
|
11 |
- sft
|
12 |
licence: license
|
13 |
---
|
14 |
|
15 |
+
# Model Card for gpt2
|
16 |
|
17 |
+
This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the [['yhavinga/mc4_nl_cleaned']](https://huggingface.co/datasets/['yhavinga/mc4_nl_cleaned']) dataset.
|
18 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
19 |
|
20 |
## Quick start
|
config.json
CHANGED
@@ -34,6 +34,6 @@
|
|
34 |
},
|
35 |
"torch_dtype": "bfloat16",
|
36 |
"transformers_version": "4.46.1",
|
37 |
-
"use_cache":
|
38 |
"vocab_size": 50257
|
39 |
}
|
|
|
34 |
},
|
35 |
"torch_dtype": "bfloat16",
|
36 |
"transformers_version": "4.46.1",
|
37 |
+
"use_cache": true,
|
38 |
"vocab_size": 50257
|
39 |
}
|