AnnyNguyen commited on
Commit
f411afd
·
verified ·
1 Parent(s): fe0bb4d

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +30 -0
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoModelForSequenceClassification": "models.ViHateT5Model",
4
+ "AutoModel": "models.ViHateT5Model",
5
+ "AutoTokenizer": "AutoTokenizer"
6
+ },
7
+ "architectures": [
8
+ "ViHateT5Model"
9
+ ],
10
+ "model_type": "t5",
11
+ "_name_or_path": "VietAI/vit5-base",
12
+ "d_model": 768,
13
+ "num_heads": 12,
14
+ "num_layers": 12,
15
+ "num_labels": 3,
16
+ "id2label": {
17
+ "0": "CLEAN",
18
+ "1": "OFFENSIVE",
19
+ "2": "HATE"
20
+ },
21
+ "label2id": {
22
+ "CLEAN": 0,
23
+ "OFFENSIVE": 1,
24
+ "HATE": 2
25
+ },
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.30.0",
28
+ "decoder_start_token_id": 0,
29
+ "pad_token_id": 0
30
+ }