add model

config.json

@@ -1,12 +1,13 @@
 {
-  "_name_or_path": "/tmp/hf_demo_may_11th",
+  "_name_or_path": "CXR-BERT-specialized-HD_c01b28b3-c621-4c74-8f70-a1cf88137dc9_0",
   "architectures": [
     "CXRBertModel"
   ],
   "attention_probs_dropout_prob": 0.25,
   "auto_map": {
     "AutoConfig": "configuration_cxrbert.CXRBertConfig",
-    "AutoModel": "modeling_cxrbert.CXRBertModel"
+    "AutoModel": "modeling_cxrbert.CXRBertModel",
+    "AutoModelForMaskedLM": "modeling_cxrbert.CXRBertModel"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,

configuration_cxrbert.py

@@ -9,6 +9,11 @@ from transformers import BertConfig
 
 
 class CXRBertConfig(BertConfig):
+    """
+    Config class for CXR-BERT model.
+    :param projection_size: Dimensionality of the joint latent space.
+    """
+
     model_type = "cxr-bert"
 
     def __init__(self, projection_size: int = 128, **kwargs: Any) -> None: