| { |
| "transformers_version": "5.5.0", |
| "architectures": [ |
| "BertForTokenClassification" |
| ], |
| "output_hidden_states": false, |
| "return_dict": true, |
| "dtype": "bfloat16", |
| "chunk_size_feed_forward": 0, |
| "is_encoder_decoder": false, |
| "id2label": { |
| "0": "B-protein", |
| "1": "B-protein_complex", |
| "2": "B-protein_enum", |
| "3": "B-protein_familiy_or_group", |
| "4": "B-protein_variant", |
| "5": "I-protein", |
| "6": "I-protein_complex", |
| "7": "I-protein_enum", |
| "8": "I-protein_familiy_or_group", |
| "9": "I-protein_variant", |
| "10": "O" |
| }, |
| "label2id": { |
| "B-protein": 0, |
| "B-protein_complex": 1, |
| "B-protein_enum": 2, |
| "B-protein_familiy_or_group": 3, |
| "B-protein_variant": 4, |
| "I-protein": 5, |
| "I-protein_complex": 6, |
| "I-protein_enum": 7, |
| "I-protein_familiy_or_group": 8, |
| "I-protein_variant": 9, |
| "O": 10 |
| }, |
| "problem_type": null, |
| "vocab_size": 30522, |
| "hidden_size": 1024, |
| "num_hidden_layers": 24, |
| "num_attention_heads": 16, |
| "intermediate_size": 4096, |
| "hidden_act": "gelu", |
| "hidden_dropout_prob": 0.2, |
| "attention_probs_dropout_prob": 0.2, |
| "max_position_embeddings": 512, |
| "type_vocab_size": 2, |
| "initializer_range": 0.02, |
| "layer_norm_eps": 1e-07, |
| "pad_token_id": 0, |
| "use_cache": true, |
| "classifier_dropout": 0.2, |
| "is_decoder": false, |
| "add_cross_attention": false, |
| "bos_token_id": null, |
| "eos_token_id": null, |
| "tie_word_embeddings": true, |
| "_name_or_path": "OpenMed/OpenMed-NER-ProteinDetect-MultiMed-335M", |
| "gradient_checkpointing": false, |
| "model_type": "bert", |
| "position_embedding_type": "absolute", |
| "output_attentions": false, |
| "_mlx_task": "token-classification", |
| "_mlx_family": "bert", |
| "_mlx_position_offset": 0, |
| "_mlx_model_type": "bert", |
| "num_labels": 11, |
| "_mlx_weights_format": "safetensors" |
| } |
