config.json

{
 "architectures": [
  "TinyGuardClassifier"
 ],
 "model_type": "tinyguard",
 "hidden_size": 256,
 "num_hidden_layers": 6,
 "num_attention_heads": 4,
 "intermediate_size": 1024,
 "vocab_size": 16384,
 "max_position_embeddings": 512,
 "num_classes": 3,
 "pad_token_id": 0,
 "framework": "mlx",
 "license": "apache-2.0",
 "id2label": {
  "0": "benign",
  "1": "direct",
  "2": "indirect"
 },
 "label2id": {
  "benign": 0,
  "direct": 1,
  "indirect": 2
 }
}