sdadas
/

polish-reranker-roberta-v2

Text Classification

information-retrieval

Inference Endpoints

Model card Files Files and versions Community

sdadas commited on Sep 20

Commit

50b1a4e

•

1 Parent(s): 083cbc1

Update config.json

Files changed (1) hide show

config.json +3 -1

config.json CHANGED Viewed

@@ -6,7 +6,9 @@
   "auto_map": {
     "AutoModel": "modeling_roberta.RobertaModel",
     "AutoModelForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification",
-    "RobertaForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification"
   },
   "bos_token_id": 0,
   "classifier_dropout": null,

   "auto_map": {
     "AutoModel": "modeling_roberta.RobertaModel",
     "AutoModelForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification",
+    "RobertaForSequenceClassification": "modeling_roberta.RobertaForSequenceClassification",
+    "AutoConfig": "modeling_roberta.RobertaConfig",
+    "RobertaConfig": "modeling_roberta.RobertaConfig"
   },
   "bos_token_id": 0,
   "classifier_dropout": null,