jinaai
/

jina-reranker-v1-tiny-en

Text Classification

Transformers.js

feature-extraction

🇪🇺 Region: EU

Model card Files Files and versions Community

Update config.json

#8

by JKONGTCHEU - opened Jul 25, 2024

base: refs/heads/main

←

from: refs/pr/8

Discussion Files changed

Files changed (1) hide show

config.json +3 -6

config.json CHANGED Viewed

@@ -1,15 +1,12 @@
 {
   "_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"
   },
   "classifier_dropout": null,
   "emb_pooler": "mean",