Update with own trained weights for version 4

config.json

@@ -15,55 +15,71 @@
   "hidden_size": 768,
   "id2label": {
     "0": "O",
-    "1": "B-SELLER_NAME",
-    "2": "I-SELLER_NAME",
-    "3": "B-SELLER_ADDRESS",
-    "4": "I-SELLER_ADDRESS",
-    "5": "B-DATE",
-    "6": "I-DATE",
-    "7": "B-NUMBER",
-    "8": "I-NUMBER",
-    "9": "B-BUYER_NAME",
-    "10": "I-BUYER_NAME",
-    "11": "B-BUYER_ADDRESS",
-    "12": "I-BUYER_ADDRESS",
-    "13": "B-TOTAL",
-    "14": "I-TOTAL",
-    "15": "B-SUBTOTAL",
-    "16": "I-SUBTOTAL",
+    "1": "B-INVOICE_NO",
+    "2": "I-INVOICE_NO",
+    "3": "B-DATE",
+    "4": "I-DATE",
+    "5": "B-CURRENCY",
+    "6": "I-CURRENCY",
+    "7": "B-SELLER_NAME",
+    "8": "I-SELLER_NAME",
+    "9": "B-SELLER_ADDRESS",
+    "10": "I-SELLER_ADDRESS",
+    "11": "B-BUYER_NAME",
+    "12": "I-BUYER_NAME",
+    "13": "B-BUYER_ADDRESS",
+    "14": "I-BUYER_ADDRESS",
+    "15": "B-TOTAL",
+    "16": "I-TOTAL",
     "17": "B-TAX",
     "18": "I-TAX",
-    "19": "B-CURRENCY",
-    "20": "I-CURRENCY",
-    "21": "B-TABLE",
-    "22": "I-TABLE"
+    "19": "B-ITEM_DESC",
+    "20": "I-ITEM_DESC",
+    "21": "B-ITEM_QTY",
+    "22": "I-ITEM_QTY",
+    "23": "B-ITEM_PRICE",
+    "24": "I-ITEM_PRICE",
+    "25": "B-ITEM_TOTAL",
+    "26": "I-ITEM_TOTAL",
+    "27": "B-BANK_NAME",
+    "28": "I-BANK_NAME",
+    "29": "B-BANK_ACCT",
+    "30": "I-BANK_ACCT"
   },
   "initializer_range": 0.02,
   "input_size": 224,
   "intermediate_size": 3072,
   "label2id": {
-    "B-BUYER_ADDRESS": 11,
-    "B-BUYER_NAME": 9,
-    "B-CURRENCY": 19,
-    "B-DATE": 5,
-    "B-NUMBER": 7,
-    "B-SELLER_ADDRESS": 3,
-    "B-SELLER_NAME": 1,
-    "B-SUBTOTAL": 15,
-    "B-TABLE": 21,
+    "B-BANK_ACCT": 29,
+    "B-BANK_NAME": 27,
+    "B-BUYER_ADDRESS": 13,
+    "B-BUYER_NAME": 11,
+    "B-CURRENCY": 5,
+    "B-DATE": 3,
+    "B-INVOICE_NO": 1,
+    "B-ITEM_DESC": 19,
+    "B-ITEM_PRICE": 23,
+    "B-ITEM_QTY": 21,
+    "B-ITEM_TOTAL": 25,
+    "B-SELLER_ADDRESS": 9,
+    "B-SELLER_NAME": 7,
     "B-TAX": 17,
-    "B-TOTAL": 13,
-    "I-BUYER_ADDRESS": 12,
-    "I-BUYER_NAME": 10,
-    "I-CURRENCY": 20,
-    "I-DATE": 6,
-    "I-NUMBER": 8,
-    "I-SELLER_ADDRESS": 4,
-    "I-SELLER_NAME": 2,
-    "I-SUBTOTAL": 16,
-    "I-TABLE": 22,
+    "B-TOTAL": 15,
+    "I-BANK_ACCT": 30,
+    "I-BANK_NAME": 28,
+    "I-BUYER_ADDRESS": 14,
+    "I-BUYER_NAME": 12,
+    "I-CURRENCY": 6,
+    "I-DATE": 4,
+    "I-INVOICE_NO": 2,
+    "I-ITEM_DESC": 20,
+    "I-ITEM_PRICE": 24,
+    "I-ITEM_QTY": 22,
+    "I-ITEM_TOTAL": 26,
+    "I-SELLER_ADDRESS": 10,
+    "I-SELLER_NAME": 8,
     "I-TAX": 18,
-    "I-TOTAL": 14,
+    "I-TOTAL": 16,
     "O": 0
   },
   "layer_norm_eps": 1e-05,
@@ -84,7 +100,6 @@
   "text_embed": true,
   "transformers_version": "5.3.0",
   "type_vocab_size": 1,
-  "use_cache": false,
   "visual_embed": true,
   "vocab_size": 50265
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6b4d27b951157862526f21ea3676c7b3517238165e4943cd378be9ca974d3944
-size 503767324
+oid sha256:1e78b3ccef6fc6c9d2dc8c45f4ae3b20c40594d60cc45d9a4bd12f744a98d98c
+size 503791932

tokenizer.json

@@ -1,7 +1,21 @@
 {
   "version": "1.0",
-  "truncation": null,
-  "padding": null,
+  "truncation": {
+    "direction": "Right",
+    "max_length": 512,
+    "strategy": "LongestFirst",
+    "stride": 0
+  },
+  "padding": {
+    "strategy": {
+      "Fixed": 512
+    },
+    "direction": "Right",
+    "pad_to_multiple_of": null,
+    "pad_id": 1,
+    "pad_type_id": 0,
+    "pad_token": "<pad>"
+  },
   "added_tokens": [
     {
       "id": 0,

tokenizer_config.json

@@ -12,7 +12,7 @@
   ],
   "eos_token": "</s>",
   "errors": "replace",
-  "is_local": false,
+  "is_local": true,
   "mask_token": "<mask>",
   "model_max_length": 512,
   "only_label_first_subword": true,