helixiora
/

distilbert-another-classifier

Model card Files Files and versions

Hristo-Karagyozov commited on Feb 12

Commit

b49d532

·

verified ·

1 Parent(s): 3a3420e

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "_name_or_path": "../distilbert-prompt-classifier/checkpoint-24",
   "activation": "gelu",
   "architectures": [
-    "DistilBertForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "dim": 768,

   "_name_or_path": "../distilbert-prompt-classifier/checkpoint-24",
   "activation": "gelu",
   "architectures": [
+    "FineTunedDistilBertWithStringLabels"
   ],
   "attention_dropout": 0.1,
   "dim": 768,