rd211 commited on
Commit
bebfb1e
·
verified ·
1 Parent(s): 97445c5

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +2 -4
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "internlm/internlm2-7b-reward",
3
  "architectures": [
4
  "InternLM2ForRewardModel"
5
  ],
@@ -21,14 +20,13 @@
21
  "num_hidden_layers": 32,
22
  "num_key_value_heads": 8,
23
  "pad_token_id": 2,
24
- "pretraining_tp": 1,
25
  "reward_token_id": 92527,
26
  "rms_norm_eps": 1e-05,
27
  "rope_scaling": null,
28
  "rope_theta": 1000000,
29
  "tie_word_embeddings": false,
30
- "torch_dtype": "float32",
31
- "transformers_version": "4.47.1",
32
  "use_cache": true,
33
  "vocab_size": 92544
34
  }
 
1
  {
 
2
  "architectures": [
3
  "InternLM2ForRewardModel"
4
  ],
 
20
  "num_hidden_layers": 32,
21
  "num_key_value_heads": 8,
22
  "pad_token_id": 2,
 
23
  "reward_token_id": 92527,
24
  "rms_norm_eps": 1e-05,
25
  "rope_scaling": null,
26
  "rope_theta": 1000000,
27
  "tie_word_embeddings": false,
28
+ "torch_dtype": "float16",
29
+ "transformers_version": "4.41.2",
30
  "use_cache": true,
31
  "vocab_size": 92544
32
  }