Text Classification
Transformers
Safetensors
mistral
feature-extraction
reward_model
custom_code
text-generation-inference
lievan commited on
Commit
4ba35fd
1 Parent(s): 4f7d3e6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -2
config.json CHANGED
@@ -1,8 +1,12 @@
1
  {
2
  "_name_or_path": "openbmb/Eurus-RM-7b",
3
  "architectures": [
4
- "MistralForCausalLM"
5
  ],
 
 
 
 
6
  "bos_token_id": 1,
7
  "eos_token_id": 2,
8
  "hidden_act": "silu",
@@ -19,7 +23,7 @@
19
  "sliding_window": 4096,
20
  "tie_word_embeddings": false,
21
  "torch_dtype": "float16",
22
- "transformers_version": "4.34.0.dev0",
23
  "use_cache": true,
24
  "vocab_size": 32000
25
  }
 
1
  {
2
  "_name_or_path": "openbmb/Eurus-RM-7b",
3
  "architectures": [
4
+ "EurusRewardModel"
5
  ],
6
+ "auto_map":{
7
+ "AutoModel": "modeling_eurus_rm.EurusRewardModel"
8
+ },
9
+ "attention_dropout": 0.0,
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,
12
  "hidden_act": "silu",
 
23
  "sliding_window": 4096,
24
  "tie_word_embeddings": false,
25
  "torch_dtype": "float16",
26
+ "transformers_version": "4.38.2",
27
  "use_cache": true,
28
  "vocab_size": 32000
29
  }