| { |
| "add_cross_attention": false, |
| "architectures": [ |
| "BertForMultiLabelTokenClassification" |
| ], |
| "attention_probs_dropout_prob": 0.1, |
| "auto_map": { |
| "AutoConfig": "configuration_multilabelbert.MultiLabelBertConfig", |
| "AutoModelForTokenClassification": "modeling_multilabelbert.BertForMultiLabelTokenClassification" |
| }, |
| "bos_token_id": null, |
| "classifier_dropout": null, |
| "custom_pipelines": { |
| "multilabel-ner": { |
| "default": { |
| "model": { |
| "pt": [ |
| "jvaquet/multilabel-classification-bert", |
| "main" |
| ] |
| } |
| }, |
| "impl": "multilabel_ner.MultilabelNerPipeline", |
| "pt": [ |
| "AutoModelForTokenClassification" |
| ], |
| "type": "text" |
| } |
| }, |
| "directionality": "bidi", |
| "dtype": "float32", |
| "eos_token_id": null, |
| "gradient_checkpointing": false, |
| "hidden_act": "gelu", |
| "hidden_dropout_prob": 0.1, |
| "hidden_size": 1024, |
| "id2label": { |
| "0": "B-MISC", |
| "1": "I-MISC", |
| "2": "E-MISC", |
| "3": "S-MISC", |
| "4": "B-ORG", |
| "5": "I-ORG", |
| "6": "E-ORG", |
| "7": "S-ORG", |
| "8": "B-PER", |
| "9": "I-PER", |
| "10": "E-PER", |
| "11": "S-PER", |
| "12": "B-LOC", |
| "13": "I-LOC", |
| "14": "E-LOC", |
| "15": "S-LOC" |
| }, |
| "initializer_range": 0.02, |
| "intermediate_size": 4096, |
| "is_decoder": false, |
| "label2id": { |
| "B-LOC": 12, |
| "B-MISC": 0, |
| "B-ORG": 4, |
| "B-PER": 8, |
| "E-LOC": 14, |
| "E-MISC": 2, |
| "E-ORG": 6, |
| "E-PER": 10, |
| "I-LOC": 13, |
| "I-MISC": 1, |
| "I-ORG": 5, |
| "I-PER": 9, |
| "S-LOC": 15, |
| "S-MISC": 3, |
| "S-ORG": 7, |
| "S-PER": 11 |
| }, |
| "layer_norm_eps": 1e-12, |
| "max_position_embeddings": 512, |
| "model_type": "MultiLabelBert", |
| "num_attention_heads": 16, |
| "num_hidden_layers": 24, |
| "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", |
| "tie_word_embeddings": true, |
| "transformers_version": "5.5.3", |
| "type_vocab_size": 2, |
| "use_cache": true, |
| "vocab_size": 28996 |
| } |
|
|
