Commit
•
42b9a76
1
Parent(s):
88b9dbc
Fix 3B config error
Browse files- config.json +7 -5
config.json
CHANGED
|
@@ -12,9 +12,9 @@
|
|
| 12 |
"vocab_size": 64000,
|
| 13 |
"use_cache": true,
|
| 14 |
"init_std": 0.02,
|
| 15 |
-
"decoder_embed_dim":
|
| 16 |
-
"decoder_layers":
|
| 17 |
-
"decoder_attention_heads":
|
| 18 |
"no_scale_embedding": true,
|
| 19 |
"add_bos_token": false,
|
| 20 |
"norm_type": "simplermsnorm",
|
|
@@ -48,11 +48,13 @@
|
|
| 48 |
0,
|
| 49 |
0,
|
| 50 |
0,
|
|
|
|
|
|
|
| 51 |
0
|
| 52 |
],
|
| 53 |
-
"hidden_dim":
|
| 54 |
"linear_act_fun": "swish",
|
| 55 |
-
"glu_dim":
|
| 56 |
"bias": false,
|
| 57 |
"torch_dtype": "bfloat16",
|
| 58 |
"transformers_version": "4.32.0"
|
|
|
|
| 12 |
"vocab_size": 64000,
|
| 13 |
"use_cache": true,
|
| 14 |
"init_std": 0.02,
|
| 15 |
+
"decoder_embed_dim": 2560,
|
| 16 |
+
"decoder_layers": 32,
|
| 17 |
+
"decoder_attention_heads": 20,
|
| 18 |
"no_scale_embedding": true,
|
| 19 |
"add_bos_token": false,
|
| 20 |
"norm_type": "simplermsnorm",
|
|
|
|
| 48 |
0,
|
| 49 |
0,
|
| 50 |
0,
|
| 51 |
+
0,
|
| 52 |
+
0,
|
| 53 |
0
|
| 54 |
],
|
| 55 |
+
"hidden_dim": 2560,
|
| 56 |
"linear_act_fun": "swish",
|
| 57 |
+
"glu_dim": 6912,
|
| 58 |
"bias": false,
|
| 59 |
"torch_dtype": "bfloat16",
|
| 60 |
"transformers_version": "4.32.0"
|