rd211 commited on
Commit
55ab31e
·
verified ·
1 Parent(s): 7444a9a

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +5 -7
config.json CHANGED
@@ -1,13 +1,11 @@
1
  {
2
- "_name_or_path": "eth-dl-rewards/internlm2-7b-mod",
3
  "architectures": [
4
- "InternLM2ForSequenceClassification"
5
  ],
6
  "attn_implementation": "eager",
7
  "auto_map": {
8
  "AutoConfig": "configuration_internlm2.InternLM2Config",
9
- "AutoModel": "eth-dl-rewards/internlm2-7b-mod--modeling_internlm2.InternLM2ForSequenceClassification",
10
- "AutoModelForCausalLM": "eth-dl-rewards/internlm2-7b-mod--modeling_internlm2.InternLM2ForCausalLM"
11
  },
12
  "bias": false,
13
  "bos_token_id": 1,
@@ -22,13 +20,13 @@
22
  "num_hidden_layers": 32,
23
  "num_key_value_heads": 8,
24
  "pad_token_id": 2,
25
- "pretraining_tp": 1,
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.46.2",
32
  "use_cache": true,
33
  "vocab_size": 92544
34
  }
 
1
  {
 
2
  "architectures": [
3
+ "InternLM2ForRewardModel"
4
  ],
5
  "attn_implementation": "eager",
6
  "auto_map": {
7
  "AutoConfig": "configuration_internlm2.InternLM2Config",
8
+ "AutoModel": "modeling_internlm2.InternLM2ForRewardModel"
 
9
  },
10
  "bias": false,
11
  "bos_token_id": 1,
 
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
  }