Instructions to use Skywork/Skywork-Reward-V2-Llama-3.1-8B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Skywork/Skywork-Reward-V2-Llama-3.1-8B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Skywork/Skywork-Reward-V2-Llama-3.1-8B")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Skywork/Skywork-Reward-V2-Llama-3.1-8B") model = AutoModelForSequenceClassification.from_pretrained("Skywork/Skywork-Reward-V2-Llama-3.1-8B") - Notebooks
- Google Colab
- Kaggle
Update pipeline tag and add library name to model card
#2
by nielsr HF Staff - opened
This PR improves the model card's metadata by:
- Updating the
pipeline_tagfromtext-classificationtotext-ranking, which is more appropriate for a reward model, ensuring it appears in the correct filtering categories (https://huggingface.co/models?pipeline_tag=text-ranking). - Adding
library_name: transformersto indicate compatibility with the Hugging Face Transformers library, allowing the "Use in Transformers" widget to appear.
chrisliu298 changed pull request status to merged