AnnyNguyen commited on
Commit
ca12ce9
·
verified ·
1 Parent(s): a260df9

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +28 -0
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoModelForSequenceClassification": "models.ViSoBERTModel",
4
+ "AutoModel": "models.ViSoBERTModel",
5
+ "AutoTokenizer": "AutoTokenizer"
6
+ },
7
+ "architectures": [
8
+ "ViSoBERTModel"
9
+ ],
10
+ "model_type": "bert",
11
+ "_name_or_path": "uitnlp/visobert",
12
+ "hidden_size": 768,
13
+ "num_attention_heads": 12,
14
+ "num_hidden_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
+ }