vihate-t5-hsd / config.json
AnnyNguyen's picture
Upload config.json with huggingface_hub
f411afd verified
{
"auto_map": {
"AutoModelForSequenceClassification": "models.ViHateT5Model",
"AutoModel": "models.ViHateT5Model",
"AutoTokenizer": "AutoTokenizer"
},
"architectures": [
"ViHateT5Model"
],
"model_type": "t5",
"_name_or_path": "VietAI/vit5-base",
"d_model": 768,
"num_heads": 12,
"num_layers": 12,
"num_labels": 3,
"id2label": {
"0": "CLEAN",
"1": "OFFENSIVE",
"2": "HATE"
},
"label2id": {
"CLEAN": 0,
"OFFENSIVE": 1,
"HATE": 2
},
"torch_dtype": "float32",
"transformers_version": "4.30.0",
"decoder_start_token_id": 0,
"pad_token_id": 0
}