test_reward_model / config.json
AlekseyKorshuk's picture
Upload model
5cd7002
{
"architectures": [
"RewardModel"
],
"base_model": "EleutherAI/gpt-neo-125M",
"model_type": "reward_model",
"n_embd": 768,
"torch_dtype": "float32",
"transformers_version": "4.25.1"
}