pvcastro commited on
Commit
5a4607f
1 Parent(s): 9dc9a01

Upload BertForEmbedderCRFTokenClassification

Browse files
Files changed (2) hide show
  1. config.json +190 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,190 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/media/pedro/repositorios/models/datalawyer/ner/pedidos-transformers-crf-fp16-v5.3",
3
+ "architectures": [
4
+ "BertForEmbedderCRFTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": 0.2,
8
+ "constrain_crf": true,
9
+ "directionality": "bidi",
10
+ "dropout_task": 0.2,
11
+ "finetuning_task": "ner",
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 768,
15
+ "id2label": {
16
+ "0": "B-ATRIBUICAO",
17
+ "1": "I-ATRIBUICAO",
18
+ "2": "E-ATRIBUICAO",
19
+ "3": "S-ATRIBUICAO",
20
+ "4": "B-DECISAO",
21
+ "5": "I-DECISAO",
22
+ "6": "E-DECISAO",
23
+ "7": "S-DECISAO",
24
+ "8": "B-FUNCAO",
25
+ "9": "I-FUNCAO",
26
+ "10": "E-FUNCAO",
27
+ "11": "S-FUNCAO",
28
+ "12": "B-FUNDAMENTO",
29
+ "13": "I-FUNDAMENTO",
30
+ "14": "E-FUNDAMENTO",
31
+ "15": "S-FUNDAMENTO",
32
+ "16": "B-LOCAL",
33
+ "17": "I-LOCAL",
34
+ "18": "E-LOCAL",
35
+ "19": "S-LOCAL",
36
+ "20": "B-ORGANIZACAO",
37
+ "21": "I-ORGANIZACAO",
38
+ "22": "E-ORGANIZACAO",
39
+ "23": "S-ORGANIZACAO",
40
+ "24": "B-PEDIDO",
41
+ "25": "I-PEDIDO",
42
+ "26": "E-PEDIDO",
43
+ "27": "S-PEDIDO",
44
+ "28": "B-PESSOA",
45
+ "29": "I-PESSOA",
46
+ "30": "E-PESSOA",
47
+ "31": "S-PESSOA",
48
+ "32": "B-REFLEXO",
49
+ "33": "I-REFLEXO",
50
+ "34": "E-REFLEXO",
51
+ "35": "S-REFLEXO",
52
+ "36": "B-TIPO_ACAO",
53
+ "37": "I-TIPO_ACAO",
54
+ "38": "E-TIPO_ACAO",
55
+ "39": "S-TIPO_ACAO",
56
+ "40": "B-TRIBUNAL",
57
+ "41": "I-TRIBUNAL",
58
+ "42": "E-TRIBUNAL",
59
+ "43": "S-TRIBUNAL",
60
+ "44": "B-VALOR_ACORDO",
61
+ "45": "I-VALOR_ACORDO",
62
+ "46": "E-VALOR_ACORDO",
63
+ "47": "S-VALOR_ACORDO",
64
+ "48": "B-VALOR_CAUSA",
65
+ "49": "I-VALOR_CAUSA",
66
+ "50": "E-VALOR_CAUSA",
67
+ "51": "S-VALOR_CAUSA",
68
+ "52": "B-VALOR_CONDENACAO",
69
+ "53": "I-VALOR_CONDENACAO",
70
+ "54": "E-VALOR_CONDENACAO",
71
+ "55": "S-VALOR_CONDENACAO",
72
+ "56": "B-VALOR_CUSTAS",
73
+ "57": "I-VALOR_CUSTAS",
74
+ "58": "E-VALOR_CUSTAS",
75
+ "59": "S-VALOR_CUSTAS",
76
+ "60": "B-VALOR_PEDIDO",
77
+ "61": "I-VALOR_PEDIDO",
78
+ "62": "E-VALOR_PEDIDO",
79
+ "63": "S-VALOR_PEDIDO",
80
+ "64": "B-VARA",
81
+ "65": "I-VARA",
82
+ "66": "E-VARA",
83
+ "67": "S-VARA",
84
+ "68": "O"
85
+ },
86
+ "initializer_range": 0.02,
87
+ "intermediate_size": 3072,
88
+ "label2id": {
89
+ "B-ATRIBUICAO": 0,
90
+ "B-DECISAO": 4,
91
+ "B-FUNCAO": 8,
92
+ "B-FUNDAMENTO": 12,
93
+ "B-LOCAL": 16,
94
+ "B-ORGANIZACAO": 20,
95
+ "B-PEDIDO": 24,
96
+ "B-PESSOA": 28,
97
+ "B-REFLEXO": 32,
98
+ "B-TIPO_ACAO": 36,
99
+ "B-TRIBUNAL": 40,
100
+ "B-VALOR_ACORDO": 44,
101
+ "B-VALOR_CAUSA": 48,
102
+ "B-VALOR_CONDENACAO": 52,
103
+ "B-VALOR_CUSTAS": 56,
104
+ "B-VALOR_PEDIDO": 60,
105
+ "B-VARA": 64,
106
+ "E-ATRIBUICAO": 2,
107
+ "E-DECISAO": 6,
108
+ "E-FUNCAO": 10,
109
+ "E-FUNDAMENTO": 14,
110
+ "E-LOCAL": 18,
111
+ "E-ORGANIZACAO": 22,
112
+ "E-PEDIDO": 26,
113
+ "E-PESSOA": 30,
114
+ "E-REFLEXO": 34,
115
+ "E-TIPO_ACAO": 38,
116
+ "E-TRIBUNAL": 42,
117
+ "E-VALOR_ACORDO": 46,
118
+ "E-VALOR_CAUSA": 50,
119
+ "E-VALOR_CONDENACAO": 54,
120
+ "E-VALOR_CUSTAS": 58,
121
+ "E-VALOR_PEDIDO": 62,
122
+ "E-VARA": 66,
123
+ "I-ATRIBUICAO": 1,
124
+ "I-DECISAO": 5,
125
+ "I-FUNCAO": 9,
126
+ "I-FUNDAMENTO": 13,
127
+ "I-LOCAL": 17,
128
+ "I-ORGANIZACAO": 21,
129
+ "I-PEDIDO": 25,
130
+ "I-PESSOA": 29,
131
+ "I-REFLEXO": 33,
132
+ "I-TIPO_ACAO": 37,
133
+ "I-TRIBUNAL": 41,
134
+ "I-VALOR_ACORDO": 45,
135
+ "I-VALOR_CAUSA": 49,
136
+ "I-VALOR_CONDENACAO": 53,
137
+ "I-VALOR_CUSTAS": 57,
138
+ "I-VALOR_PEDIDO": 61,
139
+ "I-VARA": 65,
140
+ "O": 68,
141
+ "S-ATRIBUICAO": 3,
142
+ "S-DECISAO": 7,
143
+ "S-FUNCAO": 11,
144
+ "S-FUNDAMENTO": 15,
145
+ "S-LOCAL": 19,
146
+ "S-ORGANIZACAO": 23,
147
+ "S-PEDIDO": 27,
148
+ "S-PESSOA": 31,
149
+ "S-REFLEXO": 35,
150
+ "S-TIPO_ACAO": 39,
151
+ "S-TRIBUNAL": 43,
152
+ "S-VALOR_ACORDO": 47,
153
+ "S-VALOR_CAUSA": 51,
154
+ "S-VALOR_CONDENACAO": 55,
155
+ "S-VALOR_CUSTAS": 59,
156
+ "S-VALOR_PEDIDO": 63,
157
+ "S-VARA": 67
158
+ },
159
+ "layer_norm_eps": 1e-12,
160
+ "max_position_embeddings": 512,
161
+ "model_type": "bert",
162
+ "num_attention_heads": 12,
163
+ "num_hidden_layers": 12,
164
+ "output_past": true,
165
+ "pad_token_id": 0,
166
+ "pooler_fc_size": 768,
167
+ "pooler_num_attention_heads": 12,
168
+ "pooler_num_fc_layers": 3,
169
+ "pooler_size_per_head": 128,
170
+ "pooler_type": "first_token_transform",
171
+ "position_embedding_type": "absolute",
172
+ "quantization_config": {
173
+ "bnb_4bit_compute_dtype": "float32",
174
+ "bnb_4bit_quant_type": "fp4",
175
+ "bnb_4bit_use_double_quant": false,
176
+ "llm_int8_enable_fp32_cpu_offload": false,
177
+ "llm_int8_has_fp16_weight": false,
178
+ "llm_int8_skip_modules": null,
179
+ "llm_int8_threshold": 6.0,
180
+ "load_in_4bit": false,
181
+ "load_in_8bit": true,
182
+ "quant_method": "bitsandbytes"
183
+ },
184
+ "torch_dtype": "float16",
185
+ "transformers_version": "4.36.2",
186
+ "type_vocab_size": 2,
187
+ "use_cache": true,
188
+ "use_crf": true,
189
+ "vocab_size": 29794
190
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bb40254882abc9a2e8fd50737c30262dcaae9ea86dda3375df3b4531c845f70
3
+ size 132156484