Songnguyen263 commited on
Commit
289f5c2
·
verified ·
1 Parent(s): 0507a91

Initial model upload

Browse files
Files changed (2) hide show
  1. config.json +14 -16
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,13 +1,12 @@
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,7 +38,6 @@
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,17 +68,17 @@
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
  }
 
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
  "27": "LABEL_27"
39
  },
40
  "initializer_range": 0.02,
 
41
  "label2id": {
42
  "LABEL_0": 0,
43
  "LABEL_1": 1,
 
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.53.1",
 
 
83
  "vocab_size": 30522
84
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4b3e505e912c002e29bf83714dbcb41b28c7d2f5cd0660a56ba20395d405a59
3
- size 438038624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07443f42b93010686db7019f7e4f76ca4b05c20fc99589859ae2a07457cabcca
3
+ size 267912544