| { | |
| "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": "safe", | |
| "1": "confirm", | |
| "2": "deny" | |
| }, | |
| "label2id": { | |
| "safe": 0, | |
| "confirm": 1, | |
| "deny": 2 | |
| } | |
| } |