martincc98
/

bert_a3_2

Inference Endpoints

Model card Files Files and versions Community

martincc98 commited on Oct 4

Commit

35346c1

•

1 Parent(s): c1e8a44

Upload config

Files changed (1) hide show

config.json +1 -2

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "_name_or_path": "distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
-    "DistilBertForTokenClassification"
   ],
   "attention_dropout": 0.1,
   "dim": 768,
@@ -142,7 +142,6 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
-  "torch_dtype": "float32",
   "transformers_version": "4.44.2",
   "vocab_size": 30522
 }

   "_name_or_path": "distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
+    "DistilBertForMaskedLM"
   ],
   "attention_dropout": 0.1,
   "dim": 768,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "transformers_version": "4.44.2",
   "vocab_size": 30522
 }