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,
|