MTP layers

#26
by Treene - opened

In programming tasks, MTP (4 tokens) yields a 1.7–2x speedup for simple code in Qwen3.5 35B-MoE (the speedup is even more pronounced for dense model) and a slightly lower one for complex code; for general-purpose text, the speedup is around 1.2–1.4x.
Extremely high token generation acceleration (accept_rate > 90%) also occurs when the reasoning process involves code that the model subsequently simply copies into the final output.

Why were the MTP layers removed?
Should I try adding the MTP layers from Qwen-3.5, or has the model been retrained enough that the accept_rate would be extremely low?

I have the same issue

I crave the MTP layers, too. Hello anyone at Ornith care to chime in ? ??

Answered over in #2: grafting the 19 mtp.* tensors from Qwen/Qwen3.6-35B-A3B works β€” 70.3% acceptance at n=2 under vLLM, so the post-train does not kill it.

Repo and graft script: crucible-labs/Ornith-1.0-35B-MTP. For llama.cpp, skinnyctax's Frankenstein-MTP GGUF predates it.

Sign up or log in to comment