Update README.md
Browse files
README.md
CHANGED
|
@@ -15,8 +15,9 @@ This is the 10k steps English supervised-fine-tuning (SFT) model of GPT-J using
|
|
| 15 |
|
| 16 |
You need to add this to default config file `configs/config.yaml`
|
| 17 |
|
| 18 |
-
|
| 19 |
```
|
|
|
|
| 20 |
soda-only:
|
| 21 |
datasets:
|
| 22 |
- soda:
|
|
@@ -24,8 +25,9 @@ soda-only:
|
|
| 24 |
input_max_length: 1024
|
| 25 |
```
|
| 26 |
|
| 27 |
-
|
| 28 |
```
|
|
|
|
| 29 |
dtype: fp16
|
| 30 |
log_dir: gptj-soda
|
| 31 |
model_name: EleutherAI/gpt-j-6b
|
|
|
|
| 15 |
|
| 16 |
You need to add this to default config file `configs/config.yaml`
|
| 17 |
|
| 18 |
+
|
| 19 |
```
|
| 20 |
+
data:
|
| 21 |
soda-only:
|
| 22 |
datasets:
|
| 23 |
- soda:
|
|
|
|
| 25 |
input_max_length: 1024
|
| 26 |
```
|
| 27 |
|
| 28 |
+
|
| 29 |
```
|
| 30 |
+
gptj-chai:
|
| 31 |
dtype: fp16
|
| 32 |
log_dir: gptj-soda
|
| 33 |
model_name: EleutherAI/gpt-j-6b
|