berkeley-nest / Starling-RM-7B-alpha
Other133105Updated Jul 30, 2024
Downloads
133
Likes
105
Last updated
Jul 30, 2024
License
apache-2.0
Starling-RM-7B-alpha is a reward model trained from Llama2-7B-Chat. Following the method of training reward model in the instructGPT paper, we remove the last layer of Llama2-7B Chat, and concatenate a linear layer that outputs scalar for any pair of input prompt and response.…
Tasks & library
- transformers
Tags
- transformers
- pytorch
- llama
- reward model
- RLHF
- RLAIF
Technical specifications
ParametersNot available
Librarytransformers
Languagesen
Licenseapache-2.0
