Views
No views yet
| Model | AIME2024 | MATH500 | GPQA Diamond | LiveCodeBench V2 | Avg. |
|---|---|---|---|---|---|
| DistillQwen-ThoughtY-4B | 76.7 | 95.2 | 56.1 | 75.8 | 76.0 |
| DistillQwen-ThoughtY-8B | 76.7 | 94.6 | 62.1 | 78.1 | 77.9 |
| DistillQwen-ThoughtY-32B | 90.0 | 95.2 | 63.6 | 76.3 | 81.3 |
| OpenThinker2-32B | 76.7 | 90.8 | 64.1 | 72.5 | 76.0 |
| DistillQwen-ThoughtX-32B | 80.0 | 92.6 | 64.0 | 73.4 | 77.5 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "alibaba-pai/DistillQwen-ThoughtY-4B",
5 device_map="auto"
6)
7tokenizer = AutoTokenizer.from_pretrained(model_name)
8
9prompt = "Solve ∫x e^x dx. Show your reasoning step-by-step."
10messages = [
11 {"role": "user", "content": prompt}
12]
13text = tokenizer.apply_chat_template(
14 messages,
15 tokenize=False,
16 add_generation_prompt=True,
17 enable_thinking=True # Switches between thinking and non-thinking modes. Default is True.
18)
19inputs = tokenizer([text], return_tensors="pt").to("cuda")
20outputs = model.generate(**inputs, max_new_tokens=32768)
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))1@misc{cai2025reasoningomnithoughtlargecot,
2 title={Reasoning with OmniThought: A Large CoT Dataset with Verbosity and Cognitive Difficulty Annotations},
3 author={Wenrui Cai and Chengyu Wang and Junbing Yan and Jun Huang and Xiangzhong Fang},
4 year={2025},
5 eprint={2505.10937},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL},
8 url={https://arxiv.org/abs/2505.10937}
9}