OPTD-DeepSeek-R1-Distill-Qwen-7B-Late

This is the On-Policy Transition Distillation (OPTD) student checkpoint from the paper On-Policy Transition Distillation.

Setting Value
Initialization deepseek-ai/DeepSeek-R1-Distill-Qwen-7B
Training updates 100
Intermediate blocks (zero-based) 13 / 17 / 21
Domains math, coding
Weight format BF16 safetensors
Generation mode Native reasoning template

OPTD retains final-policy OPD and aligns teacher/student log-probability increments across the selected blocks and final output on student-generated rollouts. The practical update uses fixed final-policy top-16 candidate identities and detached transition rewards and destination-layer student probabilities.

This repository contains the full inference model, tokenizer, generation configuration, and training metadata. Optimizer state and distributed training shards are not included. checksums.json records the SHA256 of the uploaded checkpoint/tokenizer files.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_id = "ModalityDance/OPTD-DeepSeek-R1-Distill-Qwen-7B-Late"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id, torch_dtype=torch.bfloat16, device_map="auto"
)
text = tokenizer.apply_chat_template(
    [{"role": "user", "content": "Solve 2x + 5 = 17."}],
    tokenize=False, add_generation_prompt=True
)
inputs = tokenizer(text, return_tensors="pt").to(model.device)
output = model.generate(**inputs, max_new_tokens=1024, do_sample=True,
                        temperature=0.7, top_p=0.95)
print(tokenizer.decode(output[0, inputs.input_ids.shape[-1]:], skip_special_tokens=True))

Evaluation and limitations

The checkpoint is intended for research on specialist-to-student distillation. Paper comparisons require the original checkpoint, prompt mode, benchmark revisions, sample counts, and graders. A short example generation does not reproduce the paper scores. Outputs can be incorrect or follow undesired instructions; generated code should be evaluated in an isolated sandbox.

License and attribution

The underlying student and specialist models retain their respective upstream license terms. Consult the linked base model's license and notices before redistribution or deployment. No new license grant for upstream model weights is implied by this model card.

Downloads last month
-
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for ModalityDance/OPTD-DeepSeek-R1-Distill-Qwen-7B-Late

Finetuned
(312)
this model