Downloads · 30 days
16
2% of all-time downloads
openllama-13b_rm_oasst-hh
reciprocate/openllama-13b_rm_oasst-hh
openllama-13b_rm_oasst-hh is a text classification model from reciprocate. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
All-time downloads
680
Public
Repo size
51.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin25.7 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- llama
- Access
- Public
- Created
- Jun 19, 2023
- Updated
- Jun 21, 2023
- SHA
- 41f353d6
Try a prompt
Model card
From the Hugging Face model README
OpenLLama-13B for reward modeling
- Dataset: https://huggingface.co/datasets/pvduy/rm_oa_hh
- Logs: https://wandb.ai/sorry/autocrit/runs/j05t4e97?workspace=user-sorry
- Code: https://github.com/CarperAI/autocrit/blob/main/train_reward_model.py
Usage:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
ckpt = "reciprocate/openllama-13b_rm_oasst-hh"
model = AutoModelForSequenceClassification.from_pretrained(ckpt, load_in_4bit=True)
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model(**tokenizer("ASSISTANT: This sentence is a lie.", return_tensors="pt"))[0].item()
Output:
-1.626953125
- Task
- Text Classification
- Library
- transformers
- Type
- llama
- License
- apache-2.0
- Languages
- en
- Created
- Jun 19, 2023
- Updated
- Jun 21, 2023