Commit
•
d6bcaf1
1
Parent(s):
4921590
add missing model type
Browse filesThis adds the missing `model_type` to load the model with `AutoModelForSequenceClassification`
- config.json +1 -0
config.json
CHANGED
@@ -14,6 +14,7 @@
|
|
14 |
},
|
15 |
"attention_probs_dropout_prob": 0.1,
|
16 |
"hidden_act": "gelu",
|
|
|
17 |
"hidden_dropout_prob": 0.1,
|
18 |
"hidden_size": 768,
|
19 |
"initializer_range": 0.02,
|
|
|
14 |
},
|
15 |
"attention_probs_dropout_prob": 0.1,
|
16 |
"hidden_act": "gelu",
|
17 |
+
"model_type": "bert",
|
18 |
"hidden_dropout_prob": 0.1,
|
19 |
"hidden_size": 768,
|
20 |
"initializer_range": 0.02,
|