IDEA-CCNL
/

Ziya-LLaMA-7B-Reward

Text Classification

text-generation-inference

Model card Files Files and versions Community

Upload folder using huggingface_hub

#6

by yuxin - opened May 9, 2023

base: refs/heads/main

←

from: refs/pr/6

Discussion Files changed

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
     "LlamaRewardModel"
   ],
   "auto_map": {
-    "AutoModelForSequenceClassification": "rm_modeling_hf.LlamaRewardModel"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,

     "LlamaRewardModel"
   ],
   "auto_map": {
+    "AutoModelForSequenceClassification": "modeling_llama_rm.LlamaRewardModel"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,