Views
No views yet
1from transformers import AutoTokenizer, GPT2ForSequenceClassification
2tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
3model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-reward")
4inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
5with torch.no_grad():
6 logits = model(**inputs).logits