Team Ai
Modelpublic

dev-analyzer/commit-message-model

sourceHugging Faceupdated 2y agoView on Hugging Face
3likes25downloads
config.json45 linesDownload Raw Back to root
1{
2  "_name_or_path": "vinai/bertweet-base",
3  "architectures": [
4    "RobertaForSequenceClassification"
5  ],
6  "attention_probs_dropout_prob": 0.1,
7  "bos_token_id": 0,
8  "classifier_dropout": null,
9  "eos_token_id": 2,
10  "gradient_checkpointing": false,
11  "hidden_act": "gelu",
12  "hidden_dropout_prob": 0.1,
13  "hidden_size": 768,
14  "id2label": {
15    "0": "Adaptive",
16    "1": "Administrative",
17    "2": "Perfective",
18    "3": "Other",
19    "4": "Corrective"
20  },
21  "initializer_range": 0.02,
22  "intermediate_size": 3072,
23  "label2id": {
24    "Adaptive": 0,
25    "Administrative": 1,
26    "Corrective": 4,
27    "Other": 3,
28    "Perfective": 2
29  },
30  "layer_norm_eps": 1e-05,
31  "max_position_embeddings": 130,
32  "model_type": "roberta",
33  "num_attention_heads": 12,
34  "num_hidden_layers": 12,
35  "pad_token_id": 1,
36  "position_embedding_type": "absolute",
37  "problem_type": "single_label_classification",
38  "tokenizer_class": "BertweetTokenizer",
39  "torch_dtype": "float32",
40  "transformers_version": "4.38.2",
41  "type_vocab_size": 1,
42  "use_cache": true,
43  "vocab_size": 64001
44}
45