Boriscii
/

bart-large-mnli-logged

Model card Files Files and versions Community

Boriscii commited on Oct 5, 2024

Commit

5f93fa2

·

verified ·

1 Parent(s): 27182a2

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -10,7 +10,8 @@
   "attention_dropout": 0.0,
   "auto_map": {
     "AutoModel": "model/modeling_modified.py",
-    "BartForSequenceClassification": "model/modeling_modified.py"
   },
   "bos_token_id": 0,
   "classif_dropout": 0.0,

   "attention_dropout": 0.0,
   "auto_map": {
     "AutoModel": "model/modeling_modified.py",
+    "BartForSequenceClassification": "model/modeling_modified.py",
+    "AutoConfig": "transformers.BartConfig"
   },
   "bos_token_id": 0,
   "classif_dropout": 0.0,