Upload DistilBertForTokenClassification

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "Mozilla/distilbert-uncased-NER-LoRA",
+  "_name_or_path": "distilbert/distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
     "DistilBertForTokenClassification"
@@ -14,21 +14,25 @@
     "2": "I-PER",
     "3": "B-ORG",
     "4": "I-ORG",
-    "5": "B-LOC",
-    "6": "I-LOC",
-    "7": "B-MISC",
-    "8": "I-MISC"
+    "5": "B-CITY",
+    "6": "I-CITY",
+    "7": "B-STATE",
+    "8": "I-STATE",
+    "9": "B-CITYSTATE",
+    "10": "I-CITYSTATE"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "B-LOC": 5,
-    "B-MISC": 7,
+    "B-CITY": 5,
+    "B-CITYSTATE": 9,
     "B-ORG": 3,
     "B-PER": 1,
-    "I-LOC": 6,
-    "I-MISC": 8,
+    "B-STATE": 7,
+    "I-CITY": 6,
+    "I-CITYSTATE": 10,
     "I-ORG": 4,
     "I-PER": 2,
+    "I-STATE": 8,
     "O": 0
   },
   "max_position_embeddings": 512,
@@ -40,6 +44,7 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
-  "transformers_version": "4.43.4",
+  "torch_dtype": "float32",
+  "transformers_version": "4.44.2",
   "vocab_size": 30522
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20e21f99611e13f3716bda86412deb8a91460937818dbe555658a8ca7bbd170e
-size 265491548
+oid sha256:ac2824235e8befb59bc5e50f7b61fe78c86ae3e35a0cfe5b0cdfc69360020789
+size 265497700