config.json

{
  "auto_map": {
    "AutoModelForSequenceClassification": "models.BartPhoModel",
    "AutoModel": "models.BartPhoModel",
    "AutoTokenizer": "AutoTokenizer"
  },
  "architectures": [
    "BartPhoModel"
  ],
  "model_type": "bart",
  "_name_or_path": "vinai/bartpho-syllable-base",
  "d_model": 768,
  "encoder_attention_heads": 16,
  "encoder_layers": 6,
  "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"
}