Instructions to use oliversssf2/distilbert-base-uncased-rm-helpful with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use oliversssf2/distilbert-base-uncased-rm-helpful with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("oliversssf2/distilbert-base-uncased-rm-helpful", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Load the reward model with the following code
from reward_modelling.reward_model import RewardModel
rm = RewardModel.from_pretrained('oliversssf2/distilbert-base-uncased-rm-helpful')