vazish commited on
Commit
ac87d6a
1 Parent(s): 5782ce0

added new models with id2label

Browse files
config.json CHANGED
@@ -9,43 +9,43 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3",
16
- "4": "LABEL_4",
17
- "5": "LABEL_5",
18
- "6": "LABEL_6",
19
- "7": "LABEL_7",
20
- "8": "LABEL_8",
21
- "9": "LABEL_9",
22
- "10": "LABEL_10",
23
- "11": "LABEL_11",
24
- "12": "LABEL_12",
25
- "13": "LABEL_13",
26
- "14": "LABEL_14",
27
- "15": "LABEL_15",
28
- "16": "LABEL_16"
29
  },
30
  "initializer_range": 0.02,
31
  "label2id": {
32
- "LABEL_0": 0,
33
- "LABEL_1": 1,
34
- "LABEL_10": 10,
35
- "LABEL_11": 11,
36
- "LABEL_12": 12,
37
- "LABEL_13": 13,
38
- "LABEL_14": 14,
39
- "LABEL_15": 15,
40
- "LABEL_16": 16,
41
- "LABEL_2": 2,
42
- "LABEL_3": 3,
43
- "LABEL_4": 4,
44
- "LABEL_5": 5,
45
- "LABEL_6": 6,
46
- "LABEL_7": 7,
47
- "LABEL_8": 8,
48
- "LABEL_9": 9
49
  },
50
  "max_position_embeddings": 512,
51
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "CC Expiration",
13
+ "1": "CC Expiration Month",
14
+ "2": "CC Expiration Year",
15
+ "3": "CC Name",
16
+ "4": "CC Number",
17
+ "5": "CC Payment Type",
18
+ "6": "CC Security Code",
19
+ "7": "CC Type",
20
+ "8": "Confirm Password",
21
+ "9": "Email",
22
+ "10": "First Name",
23
+ "11": "Form",
24
+ "12": "Last Name",
25
+ "13": "New Password",
26
+ "14": "Other",
27
+ "15": "Phone",
28
+ "16": "Zip Code"
29
  },
30
  "initializer_range": 0.02,
31
  "label2id": {
32
+ "CC Expiration": 0,
33
+ "CC Expiration Month": 1,
34
+ "CC Expiration Year": 2,
35
+ "CC Name": 3,
36
+ "CC Number": 4,
37
+ "CC Payment Type": 5,
38
+ "CC Security Code": 6,
39
+ "CC Type": 7,
40
+ "Confirm Password": 8,
41
+ "Email": 9,
42
+ "First Name": 10,
43
+ "Form": 11,
44
+ "Last Name": 12,
45
+ "New Password": 13,
46
+ "Other": 14,
47
+ "Phone": 15,
48
+ "Zip Code": 16
49
  },
50
  "max_position_embeddings": 512,
51
  "model_type": "distilbert",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:79bd4c961dd02071df3e45692e3b196805c826224f919e2679f0cb9258fcb6bd
3
  size 541363516
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a60581b70899789b8b6da9168eaa42ec3aa0079309243cc7b8b2800d0cade963
3
  size 541363516
onnx/model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d1e34b3fe865c0b62e49c163ad10c4ab64b7210e208f1095f968c1b89c91e330
3
  size 541486659
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f5334109cb824fd71e674505df462a94bfc089ab34dc8a07589ad3d09687c59
3
  size 541486659
onnx/model_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:70533cd601f21f138c8827307da1e142238ccb5f8b915fb8c61a51b3227caf6c
3
  size 135753249
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3e004b3b90a07d776414eb6762a166af341c4dd1e17c44f5ed57779084504ab
3
  size 135753249
quantize_config.json CHANGED
@@ -4,30 +4,30 @@
4
  "per_model_config": {
5
  "model": {
6
  "op_types": [
7
- "Gather",
8
- "Relu",
9
- "Reshape",
10
- "Unsqueeze",
11
  "ReduceMean",
12
- "Pow",
13
- "Where",
14
- "Equal",
15
- "Constant",
16
- "Slice",
17
  "Erf",
18
- "Sub",
19
  "MatMul",
20
- "Shape",
 
 
 
21
  "Sqrt",
 
22
  "Cast",
23
- "Gemm",
24
  "Expand",
25
- "Softmax",
 
26
  "Add",
27
- "Concat",
 
 
28
  "Div",
29
- "Mul",
30
- "Transpose"
31
  ],
32
  "weight_type": "QInt8"
33
  }
 
4
  "per_model_config": {
5
  "model": {
6
  "op_types": [
7
+ "Shape",
 
 
 
8
  "ReduceMean",
9
+ "Mul",
 
 
 
 
10
  "Erf",
11
+ "Unsqueeze",
12
  "MatMul",
13
+ "Where",
14
+ "Softmax",
15
+ "Slice",
16
+ "Reshape",
17
  "Sqrt",
18
+ "Concat",
19
  "Cast",
20
+ "Transpose",
21
  "Expand",
22
+ "Gemm",
23
+ "Gather",
24
  "Add",
25
+ "Pow",
26
+ "Sub",
27
+ "Equal",
28
  "Div",
29
+ "Constant",
30
+ "Relu"
31
  ],
32
  "weight_type": "QInt8"
33
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7dd08b51ecf04e68f191fc1644771888d7cc109cae699c3ea3fff0c02ead62be
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:304e4ea362003d441ad0874a1a7674e66771f2e439c3ceab560f3bacf735a65f
3
  size 5240