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