jinaai
/

jina-reranker-v1-turbo-en

Text Classification

Transformers.js

feature-extraction

🇪🇺 Region: EU

Model card Files Files and versions Community

rerank auto-detection TEI/infinity

#10

by michaelfeil - opened Jul 25, 2024

base: refs/heads/main

←

from: refs/pr/10

Discussion Files changed

Files changed (1) hide show

config.json +9 -6

config.json CHANGED Viewed

@@ -1,15 +1,18 @@
 {
   "_name_or_path": "jinaai/jina-bert-implementation",
-  "architectures": ["JinaBertModel"],
   "attention_probs_dropout_prob": 0.1,
   "attn_implementation": "torch",
   "auto_map": {
     "AutoConfig": "configuration_bert.JinaBertConfig",
-    "AutoModel": "modeling_bert.JinaBertModel",
-    "AutoModelForMaskedLM": "modeling_bert.JinaBertForMaskedLM",
-    "AutoModelForQuestionAnswering": "modeling_bert.JinaBertForQuestionAnswering",
-    "AutoModelForSequenceClassification": "modeling_bert.JinaBertForSequenceClassification",
-    "AutoModelForTokenClassification": "modeling_bert.JinaBertForTokenClassification"
   },
   "classifier_dropout": null,
   "emb_pooler": "mean",

 {
   "_name_or_path": "jinaai/jina-bert-implementation",
+  "architectures": ["JinaBertForSequenceClassification"],
   "attention_probs_dropout_prob": 0.1,
   "attn_implementation": "torch",
   "auto_map": {
     "AutoConfig": "configuration_bert.JinaBertConfig",
+     "AutoModel": "modeling_bert.JinaBertModel",
+    "AutoModelForSequenceClassification": "modeling_bert.JinaBertForSequenceClassification"
+  },
+  "id2label": {
+    "0": "LABEL_RERANK_0"
+  },
+  "label2id": {
+    "LABEL_RERANK_0": 0
   },
   "classifier_dropout": null,
   "emb_pooler": "mean",