Update config.json

config.json

@@ -13,7 +13,7 @@
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
-  "model_type": "roberta",
+  "model_type": "multi_task_unixcoder",
   "num_attention_heads": 12,
   "num_hidden_layers": 9,
   "output_past": true,
@@ -24,7 +24,10 @@
   "transformers_version": "4.47.0",
   "type_vocab_size": 1,
   "use_cache": true,
-  "vocab_size": 51416,
-  "num_cwe_classes": 107,
-  "vul_class_weights": [0.546, 5.806]
+  "vocab_size": 50265,
+  "num_cwe_classes": 106,
+  "auto_map": {
+    "AutoConfig": "modeling_multi_task_unixcoder.MultiTaskUnixCoderConfig",
+    "AutoModel": "modeling_multi_task_unixcoder.MultiTaskUnixCoder"
+  }
 }