update config with gradient checkpoint flag
Browse files- config.json +2 -0
config.json
CHANGED
|
@@ -8,6 +8,8 @@
|
|
| 8 |
"fused_mlp": false,
|
| 9 |
"fused_dropout_add_ln": true,
|
| 10 |
"residual_in_fp32": true,
|
|
|
|
|
|
|
| 11 |
"pad_vocab_size_multiple": 8,
|
| 12 |
"return_hidden_state": true,
|
| 13 |
"layer": {
|
|
|
|
| 8 |
"fused_mlp": false,
|
| 9 |
"fused_dropout_add_ln": true,
|
| 10 |
"residual_in_fp32": true,
|
| 11 |
+
"checkpoint_mixer": true,
|
| 12 |
+
"checkpoint_mlp": true,
|
| 13 |
"pad_vocab_size_multiple": 8,
|
| 14 |
"return_hidden_state": true,
|
| 15 |
"layer": {
|