sdadas commited on
Commit
46cf0fa
1 Parent(s): 5423bb2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -2
config.json CHANGED
@@ -1,8 +1,12 @@
1
  {
 
 
 
2
  "attention_probs_dropout_prob": 0.1,
3
  "auto_map": {
4
  "AutoModel": "modeling_roberta.RobertaModel",
5
- "AutoModelForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification"
 
6
  },
7
  "bos_token_id": 0,
8
  "classifier_dropout": null,
@@ -30,4 +34,4 @@
30
  "type_vocab_size": 1,
31
  "use_cache": true,
32
  "vocab_size": 128001
33
- }
 
1
  {
2
+ "architectures": [
3
+ "RobertaForSequenceClassification"
4
+ ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "auto_map": {
7
  "AutoModel": "modeling_roberta.RobertaModel",
8
+ "AutoModelForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification",
9
+ "RobertaForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification"
10
  },
11
  "bos_token_id": 0,
12
  "classifier_dropout": null,
 
34
  "type_vocab_size": 1,
35
  "use_cache": true,
36
  "vocab_size": 128001
37
+ }