diff --git a/config.json b/config.json index 4f2b30f..7f0d91e 100644 --- a/config.json +++ b/config.json @@ -20,6 +20,7 @@ ], "bos_token_id": 0, "eos_token_id": 2, + "gradient_checkpointing": false, "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768,