Upload TokenClassificationPipeline

config.json

@@ -5,12 +5,11 @@
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
-  "directionality": "bidi",
   "finetuning_task": "ner",
   "gradient_checkpointing": false,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "AFTER_COMPOUND",
     "1": "BEFORE_COMPOUND",
@@ -21,7 +20,7 @@
     "6": "UNKNOWN"
   },
   "initializer_range": 0.02,
-  "intermediate_size": 4096,
+  "intermediate_size": 3072,
   "label2id": {
     "AFTER_COMPOUND": 0,
     "BEFORE_COMPOUND": 1,
@@ -34,14 +33,9 @@
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
-  "num_attention_heads": 16,
-  "num_hidden_layers": 24,
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
   "pad_token_id": 0,
-  "pooler_fc_size": 768,
-  "pooler_num_attention_heads": 12,
-  "pooler_num_fc_layers": 3,
-  "pooler_size_per_head": 128,
-  "pooler_type": "first_token_transform",
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
   "transformers_version": "4.47.1",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4f5dda24f2824195f8352963f205715c5ac7d5604b220f66e149835b817c30e1
-size 1330194164
+oid sha256:e732ea3f9db6abc5c1c31c08bb073586b0b0d4943832ebf4b0167d5bd27ace6d
+size 430923588