yuxin commited on
Commit
0c59c5e
1 Parent(s): 6dd14fb

25f1b7161b716a8a537e2ca965449c4c07dda536ae36ab9e41c61d49a29847cc

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -4,7 +4,7 @@
4
  "LlamaRewardModel"
5
  ],
6
  "auto_map": {
7
- "AutoModelForSequenceClassification": "rm_modeling_hf.LlamaRewardModel"
8
  },
9
  "bos_token_id": 1,
10
  "eos_token_id": 2,
 
4
  "LlamaRewardModel"
5
  ],
6
  "auto_map": {
7
+ "AutoModelForSequenceClassification": "modeling_llama_rm.LlamaRewardModel"
8
  },
9
  "bos_token_id": 1,
10
  "eos_token_id": 2,