|
{ |
|
"_name_or_path": "/media/pedro/repositorios/models/datalawyer/ner/pedidos-transformers-crf-fp16-v5.3", |
|
"architectures": [ |
|
"BertForEmbedderCRFTokenClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.1, |
|
"classifier_dropout": 0.2, |
|
"constrain_crf": true, |
|
"directionality": "bidi", |
|
"dropout_task": 0.2, |
|
"finetuning_task": "ner", |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.1, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "B-ATRIBUICAO", |
|
"1": "I-ATRIBUICAO", |
|
"2": "E-ATRIBUICAO", |
|
"3": "S-ATRIBUICAO", |
|
"4": "B-DECISAO", |
|
"5": "I-DECISAO", |
|
"6": "E-DECISAO", |
|
"7": "S-DECISAO", |
|
"8": "B-FUNCAO", |
|
"9": "I-FUNCAO", |
|
"10": "E-FUNCAO", |
|
"11": "S-FUNCAO", |
|
"12": "B-FUNDAMENTO", |
|
"13": "I-FUNDAMENTO", |
|
"14": "E-FUNDAMENTO", |
|
"15": "S-FUNDAMENTO", |
|
"16": "B-LOCAL", |
|
"17": "I-LOCAL", |
|
"18": "E-LOCAL", |
|
"19": "S-LOCAL", |
|
"20": "B-ORGANIZACAO", |
|
"21": "I-ORGANIZACAO", |
|
"22": "E-ORGANIZACAO", |
|
"23": "S-ORGANIZACAO", |
|
"24": "B-PEDIDO", |
|
"25": "I-PEDIDO", |
|
"26": "E-PEDIDO", |
|
"27": "S-PEDIDO", |
|
"28": "B-PESSOA", |
|
"29": "I-PESSOA", |
|
"30": "E-PESSOA", |
|
"31": "S-PESSOA", |
|
"32": "B-REFLEXO", |
|
"33": "I-REFLEXO", |
|
"34": "E-REFLEXO", |
|
"35": "S-REFLEXO", |
|
"36": "B-TIPO_ACAO", |
|
"37": "I-TIPO_ACAO", |
|
"38": "E-TIPO_ACAO", |
|
"39": "S-TIPO_ACAO", |
|
"40": "B-TRIBUNAL", |
|
"41": "I-TRIBUNAL", |
|
"42": "E-TRIBUNAL", |
|
"43": "S-TRIBUNAL", |
|
"44": "B-VALOR_ACORDO", |
|
"45": "I-VALOR_ACORDO", |
|
"46": "E-VALOR_ACORDO", |
|
"47": "S-VALOR_ACORDO", |
|
"48": "B-VALOR_CAUSA", |
|
"49": "I-VALOR_CAUSA", |
|
"50": "E-VALOR_CAUSA", |
|
"51": "S-VALOR_CAUSA", |
|
"52": "B-VALOR_CONDENACAO", |
|
"53": "I-VALOR_CONDENACAO", |
|
"54": "E-VALOR_CONDENACAO", |
|
"55": "S-VALOR_CONDENACAO", |
|
"56": "B-VALOR_CUSTAS", |
|
"57": "I-VALOR_CUSTAS", |
|
"58": "E-VALOR_CUSTAS", |
|
"59": "S-VALOR_CUSTAS", |
|
"60": "B-VALOR_PEDIDO", |
|
"61": "I-VALOR_PEDIDO", |
|
"62": "E-VALOR_PEDIDO", |
|
"63": "S-VALOR_PEDIDO", |
|
"64": "B-VARA", |
|
"65": "I-VARA", |
|
"66": "E-VARA", |
|
"67": "S-VARA", |
|
"68": "O" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"label2id": { |
|
"B-ATRIBUICAO": 0, |
|
"B-DECISAO": 4, |
|
"B-FUNCAO": 8, |
|
"B-FUNDAMENTO": 12, |
|
"B-LOCAL": 16, |
|
"B-ORGANIZACAO": 20, |
|
"B-PEDIDO": 24, |
|
"B-PESSOA": 28, |
|
"B-REFLEXO": 32, |
|
"B-TIPO_ACAO": 36, |
|
"B-TRIBUNAL": 40, |
|
"B-VALOR_ACORDO": 44, |
|
"B-VALOR_CAUSA": 48, |
|
"B-VALOR_CONDENACAO": 52, |
|
"B-VALOR_CUSTAS": 56, |
|
"B-VALOR_PEDIDO": 60, |
|
"B-VARA": 64, |
|
"E-ATRIBUICAO": 2, |
|
"E-DECISAO": 6, |
|
"E-FUNCAO": 10, |
|
"E-FUNDAMENTO": 14, |
|
"E-LOCAL": 18, |
|
"E-ORGANIZACAO": 22, |
|
"E-PEDIDO": 26, |
|
"E-PESSOA": 30, |
|
"E-REFLEXO": 34, |
|
"E-TIPO_ACAO": 38, |
|
"E-TRIBUNAL": 42, |
|
"E-VALOR_ACORDO": 46, |
|
"E-VALOR_CAUSA": 50, |
|
"E-VALOR_CONDENACAO": 54, |
|
"E-VALOR_CUSTAS": 58, |
|
"E-VALOR_PEDIDO": 62, |
|
"E-VARA": 66, |
|
"I-ATRIBUICAO": 1, |
|
"I-DECISAO": 5, |
|
"I-FUNCAO": 9, |
|
"I-FUNDAMENTO": 13, |
|
"I-LOCAL": 17, |
|
"I-ORGANIZACAO": 21, |
|
"I-PEDIDO": 25, |
|
"I-PESSOA": 29, |
|
"I-REFLEXO": 33, |
|
"I-TIPO_ACAO": 37, |
|
"I-TRIBUNAL": 41, |
|
"I-VALOR_ACORDO": 45, |
|
"I-VALOR_CAUSA": 49, |
|
"I-VALOR_CONDENACAO": 53, |
|
"I-VALOR_CUSTAS": 57, |
|
"I-VALOR_PEDIDO": 61, |
|
"I-VARA": 65, |
|
"O": 68, |
|
"S-ATRIBUICAO": 3, |
|
"S-DECISAO": 7, |
|
"S-FUNCAO": 11, |
|
"S-FUNDAMENTO": 15, |
|
"S-LOCAL": 19, |
|
"S-ORGANIZACAO": 23, |
|
"S-PEDIDO": 27, |
|
"S-PESSOA": 31, |
|
"S-REFLEXO": 35, |
|
"S-TIPO_ACAO": 39, |
|
"S-TRIBUNAL": 43, |
|
"S-VALOR_ACORDO": 47, |
|
"S-VALOR_CAUSA": 51, |
|
"S-VALOR_CONDENACAO": 55, |
|
"S-VALOR_CUSTAS": 59, |
|
"S-VALOR_PEDIDO": 63, |
|
"S-VARA": 67 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"max_position_embeddings": 512, |
|
"model_type": "bert", |
|
"num_attention_heads": 12, |
|
"num_hidden_layers": 12, |
|
"output_past": true, |
|
"pad_token_id": 0, |
|
"pooler_fc_size": 768, |
|
"pooler_num_attention_heads": 12, |
|
"pooler_num_fc_layers": 3, |
|
"pooler_size_per_head": 128, |
|
"pooler_type": "first_token_transform", |
|
"position_embedding_type": "absolute", |
|
"quantization_config": { |
|
"bnb_4bit_compute_dtype": "float32", |
|
"bnb_4bit_quant_type": "fp4", |
|
"bnb_4bit_use_double_quant": false, |
|
"llm_int8_enable_fp32_cpu_offload": false, |
|
"llm_int8_has_fp16_weight": false, |
|
"llm_int8_skip_modules": null, |
|
"llm_int8_threshold": 6.0, |
|
"load_in_4bit": false, |
|
"load_in_8bit": true, |
|
"quant_method": "bitsandbytes" |
|
}, |
|
"torch_dtype": "float16", |
|
"transformers_version": "4.36.2", |
|
"type_vocab_size": 2, |
|
"use_cache": true, |
|
"use_crf": true, |
|
"vocab_size": 29794 |
|
} |
|
|