Eklav-9B-Math / README.md
AdarshSingh7647's picture
Rebrand to Eklav: rewrite model card
8dbf153 verified
|
Raw History Blame Contribute Delete
1.69 kB
metadata
base_model: zai-org/GLM-Z1-9B-0414
library_name: transformers
tags:
  - eklav
  - math-reasoning
  - cot-distillation
  - reasoning

Eklav-9B-Math

Eklav trains a model to pick up a teacher's reasoning mid thought rather than imitate it end to end. The student sees a partial reasoning trace from the teacher, with the answer revealing tail removed, and learns to continue reasoning and produce the answer on its own. The model's own reasoning is conditioned on the teacher's partial trace during training rather than trained to reproduce it word for word. Same base model, same training data as standard full trace CoT distillation, only the training objective changes.

Highlights

  • -0.7% average pass@1 across 6 math benchmarks vs. standard full trace CoT SFT, same base model and training data (MMLU excluded, general knowledge check not a math benchmark)

Model details

Base model zai-org/GLM-Z1-9B-0414
Task Math reasoning
Training method Eklav (hint conditioned SFT)
Format Merged bf16 checkpoint

Results

Pass@1 (%), single evaluation run per benchmark.

Benchmark AIME 2025 AIME 1983-2024 GPQA-Diamond GSM8K MATH-500 Omni-MATH MMLU
CotGen 35.2 59.3 49.8 95.5 91.2 39.8 81.0
Eklav 29.0 60.7 49.1 95.3 92.5 41.6 81.0

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "AdarshSingh7647/Eklav-9B-Math"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto")