Upload LiltForTokenClassification

config.json

@@ -12,32 +12,48 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "SBN",
-    "1": "address",
-    "2": "attorney_name",
-    "3": "attorneys_for",
-    "4": "b-parties",
-    "5": "email",
-    "6": "fax",
-    "7": "i-parties",
-    "8": "lawfirm",
-    "9": "other",
-    "10": "telephone"
+    "0": "B-ADDRESS",
+    "1": "B-ATTORNEY",
+    "2": "B-ATTORNEY_FOR",
+    "3": "B-EMAIL",
+    "4": "B-FAX",
+    "5": "B-LAWFIRM",
+    "6": "B-PARTY",
+    "7": "B-PHONE",
+    "8": "B-SBN",
+    "9": "I-ADDRESS",
+    "10": "I-ATTORNEY",
+    "11": "I-ATTORNEY_FOR",
+    "12": "I-EMAIL",
+    "13": "I-FAX",
+    "14": "I-LAWFIRM",
+    "15": "I-PARTY",
+    "16": "I-PHONE",
+    "17": "I-SBN",
+    "18": "OTHER"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "SBN": 0,
-    "address": 1,
-    "attorney_name": 2,
-    "attorneys_for": 3,
-    "b-parties": 4,
-    "email": 5,
-    "fax": 6,
-    "i-parties": 7,
-    "lawfirm": 8,
-    "other": 9,
-    "telephone": 10
+    "B-ADDRESS": 0,
+    "B-ATTORNEY": 1,
+    "B-ATTORNEY_FOR": 2,
+    "B-EMAIL": 3,
+    "B-FAX": 4,
+    "B-LAWFIRM": 5,
+    "B-PARTY": 6,
+    "B-PHONE": 7,
+    "B-SBN": 8,
+    "I-ADDRESS": 9,
+    "I-ATTORNEY": 10,
+    "I-ATTORNEY_FOR": 11,
+    "I-EMAIL": 12,
+    "I-FAX": 13,
+    "I-LAWFIRM": 14,
+    "I-PARTY": 15,
+    "I-PHONE": 16,
+    "I-SBN": 17,
+    "OTHER": 18
   },
   "layer_norm_eps": 1e-05,
   "max_2d_position_embeddings": 1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2bd2db014c3f8a2e840036666e6771ba299acf06206f3031f4777f1a255fb904
-size 520739868
+oid sha256:d34144012e265eb4b191fd95ed1b2f89da059f4857fbb96affa1fa92b6cd9a8d
+size 520764476