Update config.json

config.json

@@ -1,24 +1,12 @@
 {
-  "pretrained_model_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",
+  "text_config": {"pretrained_model_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext"},
   "architectures": ["ProtoForMultiLabelClassification"],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
+  "model_type": "proto",
   "num_labels": 519,
   "reduce_hidden_size": 256,
-  "hidden_size": 256,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 514,
-  "model_type": "proto",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
-  "transformers_version": "4.45.2",
-  "type_vocab_size": 1,
-  "use_cache": true,
-  "vocab_size": 250002,
-}
+  "use_attention": true,
+  "use_sigmoid": true,
+  "label_order_path": "label_order.json",
+  "prototype_vector_path": "prototype_vectors",
+  "attention_vector_path": "attention_vectors"
+}