Text Classification
Transformers
Safetensors
English
Turkish
astraeaguard_v2
prompt-injection
security
binary-classification
embeddinggemma
mixture-of-experts
custom_code
endpoints-template
Instructions to use selimk92/AstraeaGuard-V2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use selimk92/AstraeaGuard-V2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="selimk92/AstraeaGuard-V2", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained("selimk92/AstraeaGuard-V2", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Download config.json from selimk92/AstraeaGuard-V2: direct link, hf CLI and curl.
- Browser
- Download file 927 Bytes
-
https://huggingface.co/selimk92/AstraeaGuard-V2/resolve/main/config.json
- Command line
-
hf download hf://selimk92/AstraeaGuard-V2/config.json
-
curl -L -o config.json https://huggingface.co/selimk92/AstraeaGuard-V2/resolve/main/config.json
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 | |
| } | |