Text Classification
Transformers
Safetensors
PyTorch
deberta-v2
facebook
meta
llama
llama4
safety
text-embeddings-inference
Instructions to use meta-llama/Llama-Prompt-Guard-2-86M with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use meta-llama/Llama-Prompt-Guard-2-86M with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="meta-llama/Llama-Prompt-Guard-2-86M")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-Prompt-Guard-2-86M") model = AutoModelForSequenceClassification.from_pretrained("meta-llama/Llama-Prompt-Guard-2-86M") - Inference
- Notebooks
- Google Colab
- Kaggle
id2label missing in config.json?
#4
by vrdn23 - opened
It seems the id2label and label2id fields are not present in the config.json.
Thanks again for putting in the work to make this open-source!
try to add follow fields to config.json:
"id2label": {
"0": "BENIGN",
"1": "MALICIOUS"
},