microsoft
/

BiomedVLP-CXR-BERT-specialized

Fill-Mask Transformers PyTorch English cxr-bert feature-extraction exbert custom_code

Model card Files Files and versions Community

Ozan Oktay commited on May 11, 2022

Commit

12051ed

•

1 Parent(s): 0fdcbd3

add model

Files changed (2) hide show

config.json +1 -1
modeling_cxrbert.py +1 -1

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
   "attention_probs_dropout_prob": 0.25,
   "auto_map": {
     "AutoConfig": "configuration_cxrbert.CXRBertConfig",
-    "AutoModelForMaskedLM": "modeling_cxrbert.CXRBertModel"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,

   "attention_probs_dropout_prob": 0.25,
   "auto_map": {
     "AutoConfig": "configuration_cxrbert.CXRBertConfig",
+    "AutoModel": "modeling_cxrbert.CXRBertModel"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,

modeling_cxrbert.py CHANGED Viewed

@@ -12,7 +12,7 @@ from torch import Tensor as T
 from transformers import BertForMaskedLM
 from transformers.modeling_outputs import ModelOutput
-from .configuration_cxrbert import CXRBertConfig
 BERTTupleOutput = Tuple[T, T, T, T, T]

 from transformers import BertForMaskedLM
 from transformers.modeling_outputs import ModelOutput
+from .configuration_cxrbert import CXRBertConfig, CXRBertTokenizer
 BERTTupleOutput = Tuple[T, T, T, T, T]