Ray2333
/

GRM-Gemma-2B-rewardmodel-ft

Model card Files Files and versions Community

Ray2333 commited on 8 days ago

Commit

65afc19

•

1 Parent(s): 924a6ed

Update config.json

Files changed (1) hide show

config.json +1 -5

config.json CHANGED Viewed

@@ -1,14 +1,10 @@
 {
-  "_name_or_path": "/home/ry21/robust_rm/reward_models_unified_merge/GRM-Gemma-2B-sftreg_GRM-gemma-2b_baseline_80Kdata_length3000_fulltrain_lr5e-6_5e-06_cpt100",
   "architectures": [
     "GemmaForSequenceClassification"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
-  "auto_map": {
-    "AutoModel": "Ray2333/GRM-Gemma-2B-sftreg--model.AutoModelForCausalLMWithValueHead",
-    "AutoModelForSequenceClassification": "Ray2333/GRM-Gemma-2B-sftreg--model.AutoModelForCausalLMWithValueHead"
-  },
   "bos_token_id": 2,
   "eos_token_id": 1,
   "head_dim": 256,

 {
+  "_name_or_path": "Ray2333/Gemma-2B-rewardmodel-ft",
   "architectures": [
     "GemmaForSequenceClassification"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 2,
   "eos_token_id": 1,
   "head_dim": 256,