Fill-Mask
Transformers
PyTorch
Safetensors
bert
custom_code
robinzixuan commited on
Commit
bcf2123
1 Parent(s): ade9c6c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -18,10 +18,16 @@
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
 
21
  "torch_dtype": "float32",
22
  "transformers_version": "4.31.0",
23
  "type_vocab_size": 2,
24
  "use_cache": true,
25
  "vocab_size": 30522,
26
  "attn_implementation":"OutEffHop",
 
 
 
 
 
27
  }
 
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
+
22
  "torch_dtype": "float32",
23
  "transformers_version": "4.31.0",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
  "vocab_size": 30522,
27
  "attn_implementation":"OutEffHop",
28
+ "auto_map":
29
+ {"AutoConfig": "configuration_bert.BERTConfig",
30
+ "AutoModel": "modeling_bert.BertModel",
31
+ "AutoModelForMaskedLM": "modeling_opt.BertForMaskedLM"
32
+ },
33
  }