North Mini Code 1.0 DFlash

This is a research-preview auxiliary draft checkpoint for speculative decoding with CohereLabs/North-Mini-Code-1.0-w4a16. It is not a standalone language model and it cannot generate text without the North verifier/target model. It will likely also work with CohereLabs/North-Mini-Code-1.0-fp8 with slightly lower acceptance rates.

Training

This DFlash draft model was trained on 10,000 Magicoder prompts with on-policy supervision from North Mini Code.

Important Notes

  • not a base model
  • not a fine-tuning target for standalone inference
  • not a correctness guarantee
  • not byte-parity with target-only greedy decoding

Architecture

  • speculator type: DFlashDraftModel
  • draft layers: 5 Qwen3 sliding-window layers
  • hidden size: 2048
  • attention heads: 32
  • key/value heads: 4
  • head dim: 128
  • MLP intermediate size: 768
  • sliding window: 2048
  • draft vocabulary: 32,000
  • target vocabulary: 262,144
  • block size: 8
  • sample_from_anchor: false
  • auxiliary target hidden-state taps: layers [2, 24, 46]
  • tensors: 62
  • serialized tensor elements: 733,274,624
  • BF16 parameters: 732,980,480
  • selected weight hash: 0ac5964c48003f2eab08c4691357dedc11227d88302ea7c76fc47aaddab6d5b9

Largest tensors include the target embedding matrix, lm_head.weight, and the draft fc.weight projection. The tensor inventory is recorded in tensor_info.json.

Training data and supervision

  • prompt dataset: ise-uiuc/Magicoder-Evol-Instruct-110K
  • license: Apache-2.0
  • training rows: 10,000
  • selected rows: 0-499, 600-5099, and 6100-11099
  • locked holdout: rows 500-599, never trained
  • supervision: on-policy responses and target hidden states generated once from CohereLabs/North-Mini-Code-1.0-w4a16
  • response cap: 2,048 tokens
  • target layers used for supervision: [2, 24, 46]
  • later acceptance-mined continuation data: not included in this release checkpoint

Training metadata is recorded in training_metadata.json.

Matched GB10 benchmarks

These numbers come from the same matched serving contract on a GB10 (DGX Spark) machine running Marlin (which is suboptimal but predictable). Compare the rows within the same host and contract only. These absolute token/s values are not representative of maximum performance, simply a demonstration that the draft model works. :)

Contract Target-only DFlash K3 DFlash / target DSpark K4 DSpark / target
C1 39.9129 tok/s 75.5106 tok/s 1.8919x 79.7341 tok/s 1.9977x
C2 78.9486 tok/s 121.5232 tok/s 1.5393x 127.6174 tok/s 1.6165x
Model C1 EAL C1 draft acceptance C2 EAL C2 draft acceptance
DFlash K3 2.2947 43.16% 2.2856 42.85%
DSpark K4 2.4800 37.00% 2.4779 36.95%

Expected acceptance length (EAL) is the mean emitted tokens per speculative verification step, including the target bonus token.

Additional result: NVIDIA RTX 3090

DFlash was also tested on an NVIDIA RTX 3090 under the same family of matched conditions. That result is specific to DFlash; do not infer DSpark RTX 3090 portability from it.

  • target-only: 110.2714 tok/s
  • DFlash K3: 153.7188 tok/s
  • official Eagle K3: 77.9583 tok/s

Also being realistic, you wouldn't try to serve this on a single 3090 anyway the model weights leave no room for context.

Default proposal depth

Default proposal depth is K3. That is the selected deployment setting in config.json and the configuration used for the release evidence. More speculative tokens hit diminishing returns past position 3 for this DFlash drafter. You're likely to see the best speed here.

Runtime boundary

This checkpoint is served as the top-level model because config.json already embeds the verifier reference. Use it only with the companion North runtime, published at https://github.com/sdougbrown/north-mini-code-draft-runtime; this card does not describe a separate target-model --speculative_config deployment. The companion is a thin overlay on official vLLM 0.27.1 carrying PRs #49819 and #50937 (Cohere2MoE auxiliary hidden states, and an expert-bias loading fix), published as ghcr.io/sdougbrown/north-mini-code-runtime:v0.27.1-49819-50937.

export DRAFT_MODEL="${DRAFT_MODEL:?set this to the downloaded DFlash directory or Hub model ID}"
export VLLM_USE_V2_MODEL_RUNNER=1
vllm serve "${DRAFT_MODEL}" \
  --tensor-parallel-size 1 \
  --max-model-len 320000 \
  --tool-call-parser cohere_command4 \
  --tokenizer-mode cohere \
  --cohere-format cmd4 \
  --reasoning-config '{"reasoning_start_str":"<|START_THINKING|>","reasoning_end_str":"<|END_THINKING|>"}' \
  --enable-auto-tool-choice

Limitations

  • 32K draft vocabulary, not the 262,144-token target vocabulary
  • target-dependent: cannot run standalone
  • quantized greedy outputs are not guaranteed to match target-only greedy byte-for-byte
  • throughput and acceptance do not establish correctness or response quality
  • absolute throughput is host-specific

References

Files in this repo

  • README.md
  • config.json
  • config.py
  • model.safetensors
  • tensor_info.json
  • training_metadata.json
  • SHA256SUMS
  • LICENSE
  • .gitattributes
Downloads last month
-
Safetensors
Model size
0.7B params
Tensor type
I64
BF16
BOOL
Inference Providers NEW
This model isn't deployed by any Inference Provider. 馃檵 Ask for provider support

Model tree for sdougbrown/North-Mini-Code-1.0-dflash

Finetuned
(2)
this model