RLHFlow
/

RewardModel-Mistral-7B-for-DPA-v1

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Haoxiang-Wang commited on Apr 21, 2024

Commit

57271c4

·

verified ·

1 Parent(s): 12f4861

Update config.json

Files changed (1) hide show

config.json +4 -1

config.json CHANGED Viewed

@@ -1,8 +1,11 @@
 {
-  "_name_or_path": "/data-3/common/reward-cond-llm/logs/reward-model-mistral-7b_helpsteer+ultrafeedback-v2.0_instruct-v2_lr5e-6_bs256_reg3e-3/checkpoint-1092",
   "architectures": [
     "MistralForAttributePrediction"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 1,
   "eos_token_id": 2,

 {
+  "_name_or_path": "Haoxiang-Wang/RewardModel-Mistral-7B-for-DPA-v1",
   "architectures": [
     "MistralForAttributePrediction"
   ],
+  "auto_map": {
+    "AutoModelForSequenceClassification": "modeling_custom.MistralForAttributePrediction"
+  }
   "attention_dropout": 0.1,
   "bos_token_id": 1,
   "eos_token_id": 2,