system HF staff commited on
Commit
bdbd084
1 Parent(s): 869f914

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_num_labels": 2,
3
  "architectures": [
4
- "BertweetForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bad_words_ids": null,
@@ -31,7 +31,7 @@
31
  "max_length": 20,
32
  "max_position_embeddings": 130,
33
  "min_length": 0,
34
- "model_type": "bertweet",
35
  "no_repeat_ngram_size": 0,
36
  "num_attention_heads": 12,
37
  "num_beams": 1,
@@ -49,6 +49,7 @@
49
  "top_k": 50,
50
  "top_p": 1.0,
51
  "torchscript": false,
 
52
  "type_vocab_size": 1,
53
  "use_bfloat16": false,
54
  "vocab_size": 64001
1
  {
2
  "_num_labels": 2,
3
  "architectures": [
4
+ "RobertaForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bad_words_ids": null,
31
  "max_length": 20,
32
  "max_position_embeddings": 130,
33
  "min_length": 0,
34
+ "model_type": "roberta",
35
  "no_repeat_ngram_size": 0,
36
  "num_attention_heads": 12,
37
  "num_beams": 1,
49
  "top_k": 50,
50
  "top_p": 1.0,
51
  "torchscript": false,
52
+ "tokenizer_class": "BertweetTokenizer",
53
  "type_vocab_size": 1,
54
  "use_bfloat16": false,
55
  "vocab_size": 64001