jwhj commited on
Commit
e2f3276
·
verified ·
1 Parent(s): 82b0401

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -101,7 +101,7 @@ Here is a script that uses the OREO model to solve a specific math problem:
101
  from vllm import LLM, SamplingParams
102
  from transformers import AutoTokenizer
103
 
104
- model_path = "/mnt/data/ckpt/pcl/qwen_full_lr5e-6_beta0-03_rew01_actor-loss-dro_kl-reg-unbiased1e-2_plot-weights"
105
  tokenizer = AutoTokenizer.from_pretrained(model_path)
106
  llm = LLM(model_path)
107
  params = SamplingParams(temperature=0, max_tokens=2048)
 
101
  from vllm import LLM, SamplingParams
102
  from transformers import AutoTokenizer
103
 
104
+ model_path = "jwhj/Qwen2.5-Math-1.5B-OREO"
105
  tokenizer = AutoTokenizer.from_pretrained(model_path)
106
  llm = LLM(model_path)
107
  params = SamplingParams(temperature=0, max_tokens=2048)