update config with checkpointing
Browse files- config.json +2 -0
config.json
CHANGED
|
@@ -7,6 +7,8 @@
|
|
| 7 |
"embed_dropout": 0.1,
|
| 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,
|
|
|
|
| 7 |
"embed_dropout": 0.1,
|
| 8 |
"fused_mlp": false,
|
| 9 |
"fused_dropout_add_ln": true,
|
| 10 |
+
"checkpoint_mixer": true,
|
| 11 |
+
"checkpoint_mlp": true,
|
| 12 |
"residual_in_fp32": true,
|
| 13 |
"pad_vocab_size_multiple": 8,
|
| 14 |
"return_hidden_state": true,
|