EAGLE icon indicating copy to clipboard operation
EAGLE copied to clipboard

Eagle-3 for LLAMA4

Open tchaton opened this issue 8 months ago • 2 comments

Thanks for this awesome project.

It would be great to see Lllama 4 eagle-3 support ;)

Best regards, T.C

tchaton avatar Apr 23 '25 17:04 tchaton

We are working on Llama4. Please stay tuned.

hongyanz avatar Apr 24 '25 01:04 hongyanz

@hongyanz This is awesome. Super keen to try it out ! What is the ETA for it ?

Also, would it be possible to share the scripts once done. I am really keen to learn more about it.

tchaton avatar Apr 24 '25 07:04 tchaton

We have successfully trained the Eagle3 versions of Qwen3-8B and Qwen3-30B-A3B based on the official training code, and have open-sourced them. On a single H200 GPU using the sglang inference framework, Qwen3-8B with Eagle3 achieves a performance boost from 186 tokens/second to 365 tokens/second, while Qwen3-30B-A3B with Eagle3 improves from 147 tokens/second to 231 tokens/second.

We used the ultra_200k test set and re-ran inference on Qwen3 to regenerate the data, which was then used as the final training set.A total of 600K dialogues were used as the training set.

https://huggingface.co/Tengyunw/qwen3_30b_moe_eagle3

https://huggingface.co/Tengyunw/qwen3_8b_eagle3

Additionally, we have also published a report detailing how to reproduce the Eagle3 training process. The report link is provided below for your reference if needed.

https://mp.weixin.qq.com/s/Dmdg6aLgFHZEcm6TY1vKkA

https://zhuanlan.zhihu.com/p/1923763301432662012

jiahe7ay avatar Jul 02 '25 07:07 jiahe7ay

@tchaton

jiahe7ay avatar Jul 02 '25 08:07 jiahe7ay

@tchaton We have updated our Readme with multiple llama-4 EAGLE-3 head by the community.

hongyanz avatar Jul 24 '25 23:07 hongyanz