Update `config.json` to make it compatible with TEI

#13
by alvarobartt HF Staff - opened
Files changed (1) hide show
  1. config.json +7 -1
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_name_or_path": "jinaai/jina-bert-implementation",
3
- "architectures": ["JinaBertModel"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "attn_implementation": "torch",
6
  "auto_map": {
@@ -18,8 +18,14 @@
18
  "hidden_act": "gelu",
19
  "hidden_dropout_prob": 0.1,
20
  "hidden_size": 384,
 
 
 
21
  "initializer_range": 0.02,
22
  "intermediate_size": 1536,
 
 
 
23
  "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 8192,
25
  "model_type": "bert",
 
1
  {
2
  "_name_or_path": "jinaai/jina-bert-implementation",
3
+ "architectures": ["JinaBertForSequenceClassification"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "attn_implementation": "torch",
6
  "auto_map": {
 
18
  "hidden_act": "gelu",
19
  "hidden_dropout_prob": 0.1,
20
  "hidden_size": 384,
21
+ "id2label": {
22
+ "0": "LABEL_0"
23
+ },
24
  "initializer_range": 0.02,
25
  "intermediate_size": 1536,
26
+ "label2id": {
27
+ "LABEL_0": 0
28
+ },
29
  "layer_norm_eps": 1e-12,
30
  "max_position_embeddings": 8192,
31
  "model_type": "bert",