Downloads · 30 days
18
1% of all-time downloads
gpt2-rlhf-reward
sugam11/gpt2-rlhf-reward
gpt2-rlhf-reward is a text classification model from sugam11. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
All-time downloads
2.5K
Public
Repo size
2.9 GB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin1.4 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2023
- Updated
- Mar 2, 2023
- SHA
- 1c802546
Try a prompt
Datasets
Model card
From the Hugging Face model README
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
- Task
- Text Classification
- Library
- transformers
- Type
- gpt2
- License
- mit
- Created
- Mar 2, 2023
- Updated
- Mar 2, 2023