Andrazp commited on
Commit
cd48c0c
1 Parent(s): d21d032

changed config to work with automodel

Browse files
Files changed (1) hide show
  1. config.json +1 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "BertForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "finetuning_task": null,
@@ -25,7 +25,6 @@
25
  "layer_norm_eps": 1e-12,
26
  "max_position_embeddings": 512,
27
  "model_type": "bert",
28
- "model_type": "bert",
29
  "num_attention_heads": 12,
30
  "num_hidden_layers": 12,
31
  "num_labels": 4,
 
1
  {
2
  "architectures": [
3
+ "BertForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "finetuning_task": null,
 
25
  "layer_norm_eps": 1e-12,
26
  "max_position_embeddings": 512,
27
  "model_type": "bert",
 
28
  "num_attention_heads": 12,
29
  "num_hidden_layers": 12,
30
  "num_labels": 4,