selvatas commited on
Commit
26d1983
·
verified ·
1 Parent(s): b9f4913

Upload ModernBertForTokenClassification

Browse files
Files changed (2) hide show
  1. config.json +164 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ModernBertForTokenClassification"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "bos_token_id": 2,
9
+ "classifier_activation": "gelu",
10
+ "classifier_bias": false,
11
+ "classifier_dropout": 0.0,
12
+ "classifier_pooling": "mean",
13
+ "cls_token_id": 1,
14
+ "decoder_bias": true,
15
+ "deterministic_flash_attn": false,
16
+ "dtype": "float32",
17
+ "embedding_dropout": 0.0,
18
+ "eos_token_id": 1,
19
+ "global_attn_every_n_layers": 3,
20
+ "global_rope_theta": 160000,
21
+ "gradient_checkpointing": false,
22
+ "hidden_activation": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_size": 768,
25
+ "id2label": {
26
+ "0": "O",
27
+ "1": "account_number",
28
+ "2": "age",
29
+ "3": "api_key",
30
+ "4": "bank_routing_number",
31
+ "5": "biometric_identifier",
32
+ "6": "blood_type",
33
+ "7": "certificate_license_number",
34
+ "8": "city",
35
+ "9": "company_name",
36
+ "10": "coordinate",
37
+ "11": "country",
38
+ "12": "county",
39
+ "13": "credit_debit_card",
40
+ "14": "customer_id",
41
+ "15": "cvv",
42
+ "16": "date",
43
+ "17": "date_of_birth",
44
+ "18": "date_time",
45
+ "19": "device_identifier",
46
+ "20": "education_level",
47
+ "21": "email",
48
+ "22": "employee_id",
49
+ "23": "employment_status",
50
+ "24": "fax_number",
51
+ "25": "first_name",
52
+ "26": "gender",
53
+ "27": "health_plan_beneficiary_number",
54
+ "28": "http_cookie",
55
+ "29": "ipv4",
56
+ "30": "ipv6",
57
+ "31": "language",
58
+ "32": "last_name",
59
+ "33": "license_plate",
60
+ "34": "mac_address",
61
+ "35": "medical_record_number",
62
+ "36": "national_id",
63
+ "37": "occupation",
64
+ "38": "password",
65
+ "39": "phone_number",
66
+ "40": "pin",
67
+ "41": "political_view",
68
+ "42": "postcode",
69
+ "43": "race_ethnicity",
70
+ "44": "religious_belief",
71
+ "45": "sexuality",
72
+ "46": "ssn",
73
+ "47": "state",
74
+ "48": "street_address",
75
+ "49": "swift_bic",
76
+ "50": "tax_id",
77
+ "51": "time",
78
+ "52": "unique_id",
79
+ "53": "url",
80
+ "54": "user_name",
81
+ "55": "vehicle_identifier"
82
+ },
83
+ "initializer_cutoff_factor": 2.0,
84
+ "initializer_range": 0.01,
85
+ "intermediate_size": 1152,
86
+ "label2id": {
87
+ "O": 0,
88
+ "account_number": 1,
89
+ "age": 2,
90
+ "api_key": 3,
91
+ "bank_routing_number": 4,
92
+ "biometric_identifier": 5,
93
+ "blood_type": 6,
94
+ "certificate_license_number": 7,
95
+ "city": 8,
96
+ "company_name": 9,
97
+ "coordinate": 10,
98
+ "country": 11,
99
+ "county": 12,
100
+ "credit_debit_card": 13,
101
+ "customer_id": 14,
102
+ "cvv": 15,
103
+ "date": 16,
104
+ "date_of_birth": 17,
105
+ "date_time": 18,
106
+ "device_identifier": 19,
107
+ "education_level": 20,
108
+ "email": 21,
109
+ "employee_id": 22,
110
+ "employment_status": 23,
111
+ "fax_number": 24,
112
+ "first_name": 25,
113
+ "gender": 26,
114
+ "health_plan_beneficiary_number": 27,
115
+ "http_cookie": 28,
116
+ "ipv4": 29,
117
+ "ipv6": 30,
118
+ "language": 31,
119
+ "last_name": 32,
120
+ "license_plate": 33,
121
+ "mac_address": 34,
122
+ "medical_record_number": 35,
123
+ "national_id": 36,
124
+ "occupation": 37,
125
+ "password": 38,
126
+ "phone_number": 39,
127
+ "pin": 40,
128
+ "political_view": 41,
129
+ "postcode": 42,
130
+ "race_ethnicity": 43,
131
+ "religious_belief": 44,
132
+ "sexuality": 45,
133
+ "ssn": 46,
134
+ "state": 47,
135
+ "street_address": 48,
136
+ "swift_bic": 49,
137
+ "tax_id": 50,
138
+ "time": 51,
139
+ "unique_id": 52,
140
+ "url": 53,
141
+ "user_name": 54,
142
+ "vehicle_identifier": 55
143
+ },
144
+ "layer_norm_eps": 1e-06,
145
+ "local_attention": 128,
146
+ "local_rope_theta": 160000,
147
+ "mask_token_id": 4,
148
+ "max_position_embeddings": 8192,
149
+ "mlp_bias": false,
150
+ "mlp_dropout": 0.0,
151
+ "model_type": "modernbert",
152
+ "norm_bias": false,
153
+ "norm_eps": 1e-05,
154
+ "num_attention_heads": 12,
155
+ "num_hidden_layers": 22,
156
+ "pad_token_id": 0,
157
+ "position_embedding_type": "sans_pos",
158
+ "repad_logits_with_grad": false,
159
+ "sep_token_id": 1,
160
+ "sparse_pred_ignore_index": -100,
161
+ "sparse_prediction": false,
162
+ "transformers_version": "4.57.2",
163
+ "vocab_size": 256000
164
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:382d8cc66286fe8def003d35bcc9f76a4b9f5de9a56c192ef837dbd071995c94
3
+ size 1230307536