mahdin70 commited on
Commit
6d342bc
·
verified ·
1 Parent(s): f2797c6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +29 -6
config.json CHANGED
@@ -1,9 +1,32 @@
1
  {
2
- "architectures": [
3
- "MultiTaskUnixCoder"
4
- ],
 
 
 
 
 
 
 
 
 
 
 
5
  "model_type": "multi_task_unixcoder",
6
- "num_cwe_classes": 12,
 
 
 
 
7
  "torch_dtype": "float32",
8
- "transformers_version": "4.47.0"
9
- }
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "microsoft/unixcoder-base",
3
+ "architectures": ["MultiTaskUnixCoder"],
4
+ "attention_probs_dropout_prob": 0.1,
5
+ "bos_token_id": 0,
6
+ "classifier_dropout": null,
7
+ "eos_token_id": 2,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-05,
15
+ "max_position_embeddings": 514,
16
  "model_type": "multi_task_unixcoder",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 9,
19
+ "output_past": true,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.47.0",
24
+ "type_vocab_size": 1,
25
+ "use_cache": true,
26
+ "vocab_size": 50265,
27
+ "num_cwe_classes": 12,
28
+ "auto_map": {
29
+ "AutoConfig": "modeling_multi_task_unixcoder.MultiTaskUnixCoderConfig",
30
+ "AutoModel": "modeling_multi_task_unixcoder.MultiTaskUnixCoder"
31
+ }
32
+ }