AstraeaGuard-V2 / config.json
selimk92's picture
Add Transformers AutoModel integration and custom inference endpoint handler
9e9a98f verified
Raw History Blame Contribute Delete
927 Bytes
{
"model_type": "astraeaguard_v2",
"architectures": [
"AstraeaGuardForSequenceClassification"
],
"auto_map": {
"AutoConfig": "configuration_astraeaguard.AstraeaGuardConfig",
"AutoModelForSequenceClassification": "modeling_astraeaguard.AstraeaGuardForSequenceClassification"
},
"encoder_id": "google/embeddinggemma-300m",
"encoder_revision": "57c266a740f537b4dc058e1b0cda161fd15afa75",
"task_prefix": "task: classification | query: ",
"max_seq_length": 512,
"embedding_dim": 768,
"default_head": "moe",
"id2label": {
"0": "BENIGN",
"1": "ATTACK"
},
"label2id": {
"BENIGN": 0,
"ATTACK": 1
},
"tie_word_embeddings": false,
"dense_manifest_sha256": "f18158f9a6b2345aea1edae5d6ac638a0387c02b8c1ee614cb20db0a3d91a2c5",
"moe_manifest_sha256": "8b487d8948b011c24c2fb3010152e8ddc06679b133666e843b167a684d3c7814",
"encoder_frozen": true,
"production_approved": false
}