Views
No views yet
<think> tags before providing its final answer.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "empero-ai/qwen3.5-9b-opus-4.6-distill"
4tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
6
7messages = [
8 {"role": "system", "content": "You are a deep reasoning AI. When given a problem, you think through it carefully and methodically inside <think> tags before providing your final answer."},
9 {"role": "user", "content": "What is the probability of rolling at least one six in four dice rolls?"}
10]
11
12text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
13inputs = tokenizer(text, return_tensors="pt").to(model.device)
14outputs = model.generate(**inputs, max_new_tokens=2048, temperature=0.7)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))<think> tags, followed by a clear final answer:<think>
Let me work through this step by step.
The probability of NOT rolling a six on a single die is 5/6.
The probability of NOT rolling a six on any of four rolls is (5/6)^4.
(5/6)^4 = 625/1296 ≈ 0.4823
So the probability of rolling at least one six is:
1 - 625/1296 = 671/1296 ≈ 0.5177
That's just over 50% — slightly more likely than not.
</think>
The probability of rolling at least one six in four dice rolls is **671/1296**, or approximately **51.77%**.
This is calculated by finding the complement: the chance of getting no sixes at all is (5/6)⁴ ≈ 48.23%, so the chance of at least one six is 1 − (5/6)⁴ ≈ 51.77%.| Metric | Epoch 1 | Epoch 2 (best) | Epoch 3 |
|---|---|---|---|
| Eval Loss | 0.5205 | 0.4809 | 0.4915 |
| Eval Token Accuracy | 0.8494 | 0.8615 | 0.8617 |
| Eval Entropy | 0.508 | 0.434 | 0.394 |
| Dataset | Examples | Format |
|---|---|---|
| nohurry/Opus-4.6-Reasoning-3000x-filtered | 2,326 | Problem → thinking → solution |
| Jackrong/Qwen3.5-reasoning-700x | 633 | ShareGPT with <think> tags |
| TeichAI/claude-4.5-opus-high-reasoning-250x | 250 | Messages with <think> tags |
| Roman1111111/claude-opus-4.6-10000x | 9,631 | Messages with reasoning traces |
| Total | 12,840 |