MTP support?

#2
by volodXYZ - opened

Any chance that you will re-train the MTP head to match the updated weights? Cheers.

BottleCapAI org

@volodXYZ yes, it's on the top of our todo list 😊

BottleCapAI org

Please also give a try to the current MTP weights, they've worked decently for us so far.

Would you consider training DSpark speculator model too?

BottleCapAI org

@volodXYZ we validated the MTP weights and they should work as good in terms of acceptance rate and provided speedup as with the original Qwen, so I will slowly close the issue if you are ok with it :)

@vcerny for now we probably will not train additional speculators as there are a few other efforts we want to focus on, but we might consider it if it continues to be in high demand. Do you happen to have some numbers for how much better is DSpark than vanilla MTP for the original Qwen3.6-27B?

It’s much better. Both DSpark and DFlash can significantly improve inference speed. Compared with standard decoding, DFlash can deliver several times higher performance, while DSpark typically provides an additional 10% improvement on top of DFlash.

https://z-lab.ai/projects/dflash/

I can confirm MTP-4 works fine with the model.

I don't think there is enough data to evaluate MTP vs DSpark for qwen3.6 27B , but I made ChatGPT to estimate it based on results from other Qwen models and other results found online. Take it with a grain of solt...

image

BottleCapAI org
β€’
edited 1 day ago

@wano @vcerny I ran a quick test on a few prompts, the original DFlash predictor trained for Qwen still works with ThinkingCap - it has close to 2x the tok/s of the vanilla MTP, average acceptance length of 7.32 (compared to ~3.6 of classic MTP) and 5x speedup over no speculative decoding. Compared to 7.36 avg accepted len that it gets with the original Qwen, the tok/s is almost the same and ours should still be much faster with the reduced thinking.

For now I'd recommend to use that, and I'll add it to our backlog to retrain the predictor properly - but can't promise when that would happen given that the benefits seem not as significant πŸ˜€

Sign up or log in to comment