Update config.json

config.json

@@ -6,7 +6,8 @@
     "auto_map": {
         "AutoConfig": "config_codesage.CodeSageConfig",
         "AutoModel": "modeling_codesage.CodeSageModel",
-        "AutoModelForSequenceClassification": "modeling_codesage.CodeSageForSequenceClassification",
+        "AutoModelForMaskedLM": "modeling_codesage.CodeSageForMaskedLM",
+        "AutoModelForSequenceClassification": "modeling_codesage.CodeSageForSequenceClassification"
     },
     "activation_function": "gelu_new",
     "attention_dropout_prob": 0.1,
@@ -20,4 +21,4 @@
     "max_position_embeddings": 2048,
     "residual_dropout_prob": 0.1,
     "vocab_size": 49154
-}
+}