Boriscii
/

bart-large-mnli-logged

Model card Files Files and versions Community

Boriscii commited on Oct 5, 2024

Commit

e712f0d

·

verified ·

1 Parent(s): bc22ed1

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -9,8 +9,8 @@
   ],
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoModel": "model/modeling_modified.py",
-    "BartForSequenceClassification": "model/modeling_modified.py",
     "AutoConfig": "config.BartConfig"
   },
   "bos_token_id": 0,

   ],
   "attention_dropout": 0.0,
   "auto_map": {
+    "AutoModel": "modeling_bart.ModifiedBartForSequenceClassificationWithHook",
+    "BartForSequenceClassification": "modeling_bart.ModifiedBartForSequenceClassificationWithHook",
     "AutoConfig": "config.BartConfig"
   },
   "bos_token_id": 0,