from transformers import AutoTokenizer, GPT2ForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-reward")
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
with torch.no_grad():
  logits = model(**inputs).logits

Downloads last month: 12

Inference Examples

Text Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

sugam11
/

gpt2-rlhf-reward

Dataset used to train sugam11/gpt2-rlhf-reward