Babelscape
/

Qwen2.5-Math-7B-PRM800k-r

prm_reward_model

Model card Files Files and versions

rpisano commited on May 29

Commit

ae9f246

·

verified ·

1 Parent(s): a11fc3f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ arXiv: https://arxiv.org/abs/2604.17957
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModel
-repo_id = "rpisano/Qwen2.5-Math-7B-PRM800k-r"
 tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
 model = AutoModel.from_pretrained(repo_id, trust_remote_code=True).eval()
 def build_prompt(problem, steps):

 ```python
 import torch
 from transformers import AutoTokenizer, AutoModel
+repo_id = "Babelscape/Qwen2.5-Math-7B-PRM800k-r"
 tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
 model = AutoModel.from_pretrained(repo_id, trust_remote_code=True).eval()
 def build_prompt(problem, steps):