TIGER-Lab
/

AceCodeRM-7B

text-generation-inference

Model card Files Files and versions Community

WyettZ commited on Feb 4

Commit

a464e0f

·

verified ·

1 Parent(s): 73c95a1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -117,7 +117,7 @@ class Qwen2ForCausalRM(Qwen2ForCausalLM):
         else:
             return (lm_logits, loss, value)
-model_path = "CodeDPO/qwen_coder_2.5_rm"
 model = Qwen2ForCausalRM.from_pretrained(model_path, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 input_chat = [

         else:
             return (lm_logits, loss, value)
+model_path = "TIGER-Lab/AceCodeRM-7B"
 model = Qwen2ForCausalRM.from_pretrained(model_path, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 input_chat = [