yigitbekir commited on
Commit
0748ca3
·
verified ·
1 Parent(s): 3738512

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -23,6 +23,7 @@
23
  "vocab_size": 128000,
24
  "auto_map": {
25
  "AutoConfig": "configuration_bert2d.Bert2DConfig",
 
26
  "AutoModelForMaskedLM": "modeling_bert2d.Bert2DForMaskedLM",
27
  "AutoTokenizer": [
28
  "tokenization_bert2d.Bert2DTokenizer",
 
23
  "vocab_size": 128000,
24
  "auto_map": {
25
  "AutoConfig": "configuration_bert2d.Bert2DConfig",
26
+ "AutoModel": "modeling_bert2d.Bert2DModel"
27
  "AutoModelForMaskedLM": "modeling_bert2d.Bert2DForMaskedLM",
28
  "AutoTokenizer": [
29
  "tokenization_bert2d.Bert2DTokenizer",