Ozan Oktay commited on
Commit
12051ed
1 Parent(s): 0fdcbd3
Files changed (2) hide show
  1. config.json +1 -1
  2. modeling_cxrbert.py +1 -1
config.json CHANGED
@@ -6,7 +6,7 @@
6
  "attention_probs_dropout_prob": 0.25,
7
  "auto_map": {
8
  "AutoConfig": "configuration_cxrbert.CXRBertConfig",
9
- "AutoModelForMaskedLM": "modeling_cxrbert.CXRBertModel"
10
  },
11
  "classifier_dropout": null,
12
  "gradient_checkpointing": false,
 
6
  "attention_probs_dropout_prob": 0.25,
7
  "auto_map": {
8
  "AutoConfig": "configuration_cxrbert.CXRBertConfig",
9
+ "AutoModel": "modeling_cxrbert.CXRBertModel"
10
  },
11
  "classifier_dropout": null,
12
  "gradient_checkpointing": false,
modeling_cxrbert.py CHANGED
@@ -12,7 +12,7 @@ from torch import Tensor as T
12
  from transformers import BertForMaskedLM
13
  from transformers.modeling_outputs import ModelOutput
14
 
15
- from .configuration_cxrbert import CXRBertConfig
16
 
17
  BERTTupleOutput = Tuple[T, T, T, T, T]
18
 
 
12
  from transformers import BertForMaskedLM
13
  from transformers.modeling_outputs import ModelOutput
14
 
15
+ from .configuration_cxrbert import CXRBertConfig, CXRBertTokenizer
16
 
17
  BERTTupleOutput = Tuple[T, T, T, T, T]
18