This repository contains the DFlash 2 draft model for Qwen/Qwen3.6-35B-A3B. It is not a standalone language model: it runs inside a speculative decoding server and drafts tokens for the target model to verify.
DFlash 2 is a block-diffusion drafter for speculative decoding. It predicts a whole block of tokens in a single pass and keeps the top candidates at every position. A lightweight selector then traces one coherent path through them. Two-tap dynamic convolutions in the backbone keep the draft from decaying toward the end of the block. Decoding is lossless: greedy output matches the target model exactly, and sampling preserves its distribution.
Runtime: SGLang on one NVIDIA GB300, with FlashInfer for target attention and FlashAttention 4 for draft attention
Speculation block size: 8 (7 draft tokens per verification step)
Sampling: temperature 1.0, top-p 0.95, top-k 20, presence penalty 0.0, repetition penalty 1.0, with thinking enabled
Maximum new tokens: 4096
Requests: 32 per dataset and concurrency, with seed 42 for prompt selection
Prompts: benchmark formatting from z-lab/dflash; MT-Bench uses the first turn
We compare autoregressive decoding, Qwen3.6's built-in MTP with seven speculative steps, and DFlash 2. Both speculative methods propose seven draft tokens per verification step.
import os
os.environ["HF_ENDPOINT"] = "https://freeweights.allternit.com"
from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("incoai/Qwen3.6-35B-A3B-DFlash2", revision="51ef7b6923ad6c14cb1bb41c37a9041446496ab6")
model = AutoModelForCausalLM.from_pretrained("incoai/Qwen3.6-35B-A3B-DFlash2", revision="51ef7b6923ad6c14cb1bb41c37a9041446496ab6", device_map="auto")