import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "YOUR_USERNAME/YOUR_MODEL_REPO"

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float16,
    device_map="auto",
    trust_remote_code=True,
)

messages = [
    {"role": "system", "content": "You are a careful reasoning assistant."},
    {"role": "user", "content": "Solve: If 3x + 5 = 20, what is x? End with Final answer: <answer>"},
]

prompt = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True,
)

inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

with torch.inference_mode():
    outputs = model.generate(
        **inputs,
        max_new_tokens=256,
        do_sample=False,
        temperature=0.0,
        pad_token_id=tokenizer.pad_token_id,
        eos_token_id=tokenizer.eos_token_id,
    )

generated = outputs[0][inputs["input_ids"].shape[1]:]
print(tokenizer.decode(generated, skip_special_tokens=True))

## Limitations

- Performance is strongest on reasoning-style prompts close to the SFT data distribution.
- Gains are not uniform across all reasoning benchmarks.
- Some benchmark improvements may reflect output-format adaptation as well as reasoning improvement.

## Training / Eval Project

This model was trained and evaluated in a local project with raw-vs-merged comparisons on math and reasoning benchmarks including:

- GSM8K
- MATH-500
- Math-CoT-20k
- MMLU math
- BoolQ
- WinoGrande
- CommonsenseQA
- ARC-Challenge

desidero32/Qwen3.5-2B-OpenMathReasoning-Finetuned

Qwen3.5-2B SFT Merged

Model Summary

Training Data

Intended Use

Benchmark Snapshot

How to Use

Transformers

desidero32/
Qwen3.5-2B-OpenMathReasoning-Finetuned