diff --git a/config.json b/config.json index 46845a2..77524d3 100644 --- a/config.json +++ b/config.json @@ -1,5 +1,4 @@ { - "_num_labels": 3, "activation_dropout": 0.0, "activation_function": "swish", "add_bias_logits": false, @@ -10,7 +9,7 @@ "attention_dropout": 0.0, "bad_words_ids": [ [ - 58865 + 65000 ] ], "bos_token_id": 0, @@ -20,13 +19,15 @@ "decoder_ffn_dim": 2048, "decoder_layerdrop": 0.0, "decoder_layers": 6, - "decoder_start_token_id": 58865, + "decoder_start_token_id": 65000, "dropout": 0.1, "encoder_attention_heads": 8, "encoder_ffn_dim": 2048, "encoder_layerdrop": 0.0, "encoder_layers": 6, "eos_token_id": 0, + "extra_pos_embeddings": 65001, + "force_bos_token_to_be_generated": false, "id2label": { "0": "LABEL_0", "1": "LABEL_1", @@ -46,8 +47,8 @@ "normalize_embedding": false, "num_beams": 6, "num_hidden_layers": 6, - "pad_token_id": 58865, + "pad_token_id": 65000, "scale_embedding": true, "static_position_embeddings": true, - "vocab_size": 58866 + "vocab_size": 65001 }