Views
No views yet
| Model | GSM8K | MATH-500 | AIME2024 | AIME2025 |
|---|---|---|---|---|
| S1-32B | - | 93.0% | 56.7% | 26.6% |
| LIMO-32B | - | 94.8% | 57.1% | 46.6% |
| QwQ-32B | - | - | 82.1% | 70.8% |
| PromptCoT-QwQ-32B (ours) | 🔥 96.4% ± 0.2% | 🔥 96.7% ± 0.5% | 🔥 83.8% ± 2.8% | 🔥 75.4% ± 4.7% |
pip install transformers vllm torch accelerategenerate API:1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "xl-zhao/PromptCoT-QwQ-32B"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
6
7problem_statement = (
8 "A robe takes 2 bolts of blue fiber and half that much white fiber. How many bolts in total does it take?"
9)
10
11prompt = (
12 f"<|im_start|>user\n{problem_statement}\nPlease reason step by step, and put your final answer within \\boxed{{}}.<|im_end|>\n"
13 "<|im_start|>assistant\n"
14)
15
16inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
17
18with torch.no_grad():
19 output = model.generate(**inputs, max_length=32768, temperature=0.6)
20
21generated_solution = tokenizer.decode(output[0], skip_special_tokens=True)
22print(generated_solution)vLLM:1from vllm import LLM, SamplingParams
2
3model_name = "xl-zhao/PromptCoT-QwQ-32B"
4llm = LLM(model=model_name, tensor_parallel_size=1)
5
6problem_statement = (
7 "A robe takes 2 bolts of blue fiber and half that much white fiber. How many bolts in total does it take?"
8)
9
10prompt = (
11 f"<|im_start|>user\n{problem_statement}\nPlease reason step by step, and put your final answer within \\boxed{{}}.<|im_end|>\n"
12 "<|im_start|>assistant\n"
13)
14
15sampling_params = SamplingParams(temperature=0.6, max_tokens=32768)
16outputs = llm.generate([prompt], sampling_params)
17
18print(outputs[0].outputs[0].text)@article{zhao2025promptcot,
author = {Zhao, Xueliang and Wu, Wei and Guan, Jian and Kong, Lingpeng},
title = {PromptCoT: Synthesizing Olympiad-Level Problems for Mathematical Reasoning in Large Language Models},
year = {2025},
journal = {arXiv preprint arXiv:2503.02324},
url = {http://arxiv.org/abs/2503.02324}
}