diff --git a/config.json b/config.json index 37b41bc..22a3328 100644 --- a/config.json +++ b/config.json @@ -10,6 +10,7 @@ "decoder_ffn_dim": 4096, "decoder_layerdrop": 0.0, "decoder_layers": 12, + "decoder_start_token_id": 2, "do_sample": false, "dropout": 0.1, "early_stopping": true, @@ -17,9 +18,6 @@ "encoder_ffn_dim": 4096, "encoder_layerdrop": 0.0, "encoder_layers": 12, - "eos_token_ids": [ - 2 - ], "finetuning_task": null, "id2label": { "0": "LABEL_0", @@ -55,4 +53,4 @@ "torchscript": false, "use_bfloat16": false, "vocab_size": 50264 -} +} \ No newline at end of file