Instructions to use yueliu1999/GuardReasoner-1B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use yueliu1999/GuardReasoner-1B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="yueliu1999/GuardReasoner-1B")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("yueliu1999/GuardReasoner-1B", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Add pipeline tag, link to paper
#1
by nielsr HF Staff - opened
This PR adds the pipeline_tag to make sure the model can be found at https://huggingface.co/models?pipeline_tag=text-classification.
It also ensures it is linked to the paper GuardReasoner: Towards Reasoning-based LLM Safeguards.
yueliu1999 changed pull request status to merged