Medic23 commited on
Commit
cea55b4
1 Parent(s): a70c063

Add model_type to config.json

Browse files

`AutoModelForMaskedLM` fails to infer the model type from the name of the model, and cannot load this model if the `model_type` property is missing from the config file.

I assumed the correct value to be "bert" from the name of the model, apologies if it is the wrong type for this model.

Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -6,6 +6,7 @@
6
  "initializer_range": 0.02,
7
  "intermediate_size": 3072,
8
  "max_position_embeddings": 512,
 
9
  "num_attention_heads": 12,
10
  "num_hidden_layers": 12,
11
  "type_vocab_size": 2,
 
6
  "initializer_range": 0.02,
7
  "intermediate_size": 3072,
8
  "max_position_embeddings": 512,
9
+ "model_type": "bert",
10
  "num_attention_heads": 12,
11
  "num_hidden_layers": 12,
12
  "type_vocab_size": 2,