fix the bug : RuntimeError: mat1 and mat2 shapes cannot be multiplied
#1
by
zhengr
- opened
- config.json +1 -1
config.json
CHANGED
|
@@ -15,7 +15,7 @@
|
|
| 15 |
"num_hidden_layers": 40,
|
| 16 |
"num_key_value_heads": 40,
|
| 17 |
"pad_token_id": 0,
|
| 18 |
-
"pretraining_tp":
|
| 19 |
"rms_norm_eps": 1e-05,
|
| 20 |
"rope_scaling": null,
|
| 21 |
"tie_word_embeddings": false,
|
|
|
|
| 15 |
"num_hidden_layers": 40,
|
| 16 |
"num_key_value_heads": 40,
|
| 17 |
"pad_token_id": 0,
|
| 18 |
+
"pretraining_tp": 1,
|
| 19 |
"rms_norm_eps": 1e-05,
|
| 20 |
"rope_scaling": null,
|
| 21 |
"tie_word_embeddings": false,
|