SafeAILab / EAGLE

Official Implementation of EAGLE
https://arxiv.org/pdf/2406.16858
Apache License 2.0
622 stars 59 forks source link

Llama 3 support? #67

Closed joshmackwilliams closed 1 hour ago

joshmackwilliams commented 1 month ago

Is Llama 3 (specifically the instruct models) supported by the code as-is (Llama 2 path), or are changes required (e.g., to support the new tokenizer)?

And, do you have any plans to release a pretrained Eagle for Llama 3? (e.g., like https://huggingface.co/yuhuili/EAGLE-llama2-chat-7B)

Liyuhui-12 commented 1 month ago

You need to modify the instruction templates (such as those in eagle/ge_data/ge_data_all_vicuna.py). Training the draft model for LLaMA 3 is our next step.

kalradivyanshu commented 2 weeks ago

Hey @Liyuhui-12 any updates on LLaMA-3 support. Btw, thank you for all your hard work!

Liyuhui-12 commented 3 days ago

Hey @kalradivyanshu, support for LLaMA3 has now been updated.