lorenzozan commited on
Commit
125c71c
·
verified ·
1 Parent(s): ec028fe

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -9,4 +9,8 @@
9
  "pretrained_model_name": "bert-base-uncased",
10
  "torch_dtype": "float32",
11
  "transformers_version": "4.37.0"
 
 
 
 
12
  }
 
9
  "pretrained_model_name": "bert-base-uncased",
10
  "torch_dtype": "float32",
11
  "transformers_version": "4.37.0"
12
+ "auto_map": {
13
+ "AutoConfig": "configuration_me2bert.ME2BertConfig",
14
+ "AutoModel": "modeling_me2bert.ME2BertModel"
15
+ }
16
  }