michaelfeil commited on
Commit
7f00719
1 Parent(s): 0d03be9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -5,6 +5,7 @@
5
  "attn_implementation": "torch",
6
  "auto_map": {
7
  "AutoConfig": "configuration_bert.JinaBertConfig",
 
8
  "AutoModelForSequenceClassification": "modeling_bert.JinaBertForSequenceClassification"
9
  },
10
  "id2label": {
 
5
  "attn_implementation": "torch",
6
  "auto_map": {
7
  "AutoConfig": "configuration_bert.JinaBertConfig",
8
+ "AutoModel": "modeling_bert.JinaBertModel",
9
  "AutoModelForSequenceClassification": "modeling_bert.JinaBertForSequenceClassification"
10
  },
11
  "id2label": {