Songnguyen263 commited on
Commit
7434d31
·
verified ·
1 Parent(s): 9268bc3

Initial model upload

Browse files
Files changed (2) hide show
  1. config.json +15 -13
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,12 +1,13 @@
1
  {
2
- "activation": "gelu",
3
  "architectures": [
4
- "DistilBertForSequenceClassification"
5
  ],
6
- "attention_dropout": 0.1,
7
- "dim": 768,
8
- "dropout": 0.1,
9
- "hidden_dim": 3072,
 
 
10
  "id2label": {
11
  "0": "LABEL_0",
12
  "1": "LABEL_1",
@@ -38,6 +39,7 @@
38
  "27": "LABEL_27"
39
  },
40
  "initializer_range": 0.02,
 
41
  "label2id": {
42
  "LABEL_0": 0,
43
  "LABEL_1": 1,
@@ -68,17 +70,17 @@
68
  "LABEL_8": 8,
69
  "LABEL_9": 9
70
  },
 
71
  "max_position_embeddings": 512,
72
- "model_type": "distilbert",
73
- "n_heads": 12,
74
- "n_layers": 6,
75
  "pad_token_id": 0,
 
76
  "problem_type": "multi_label_classification",
77
- "qa_dropout": 0.1,
78
- "seq_classif_dropout": 0.2,
79
- "sinusoidal_pos_embds": false,
80
- "tie_weights_": true,
81
  "torch_dtype": "float32",
82
  "transformers_version": "4.52.4",
 
 
83
  "vocab_size": 30522
84
  }
 
1
  {
 
2
  "architectures": [
3
+ "BertForSequenceClassification"
4
  ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
  "id2label": {
12
  "0": "LABEL_0",
13
  "1": "LABEL_1",
 
39
  "27": "LABEL_27"
40
  },
41
  "initializer_range": 0.02,
42
+ "intermediate_size": 3072,
43
  "label2id": {
44
  "LABEL_0": 0,
45
  "LABEL_1": 1,
 
70
  "LABEL_8": 8,
71
  "LABEL_9": 9
72
  },
73
+ "layer_norm_eps": 1e-12,
74
  "max_position_embeddings": 512,
75
+ "model_type": "bert",
76
+ "num_attention_heads": 12,
77
+ "num_hidden_layers": 12,
78
  "pad_token_id": 0,
79
+ "position_embedding_type": "absolute",
80
  "problem_type": "multi_label_classification",
 
 
 
 
81
  "torch_dtype": "float32",
82
  "transformers_version": "4.52.4",
83
+ "type_vocab_size": 2,
84
+ "use_cache": true,
85
  "vocab_size": 30522
86
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:06b070309d07ace1bc86fde2f660a444ff59a6e23f694325af5a372ee9474d83
3
- size 267912544
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4b3e505e912c002e29bf83714dbcb41b28c7d2f5cd0660a56ba20395d405a59
3
+ size 438038624