hendrydong/preference_700K
Viewer • Updated • 700k • 793 • 17
How to use yyqoni/rlhflow-llama-3-sft-8b-v2-token-rm-700k with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="yyqoni/rlhflow-llama-3-sft-8b-v2-token-rm-700k") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("yyqoni/rlhflow-llama-3-sft-8b-v2-token-rm-700k")
model = AutoModelForSequenceClassification.from_pretrained("yyqoni/rlhflow-llama-3-sft-8b-v2-token-rm-700k")This is the token-wise reward model introduced in the preprint Segmenting Text and Learning Their Rewards for Improved RLHF in Language Models (https://arxiv.org/abs/2501.02790). For more details, please visit our repository at https://github.com/yinyueqin/DenseRewardRLHF-PPO.
Base model
RLHFlow/LLaMA3-SFT-v2