-
-
Notifications
You must be signed in to change notification settings - Fork 6.2k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
[Misc][MoE] add Deepseek-V3 moe tuning support #12558
[Misc][MoE] add Deepseek-V3 moe tuning support #12558
Conversation
Bypass graph capture in the moe tuning script for deepseek-v3 Signed-off-by: Divakar Verma <divakar.verma@amd.com>
👋 Hi! Thank you for contributing to the vLLM project. Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging. To run CI, PR reviewers can do one of these:
🚀 |
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
6e40b95
to
edda3cb
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you for the contribution. This looks good to me.
Note: I ran into a couple of somewhat orthogonal issues when trying it out, so threw up #12562 to address them
LGTM for this part, but we still need to add support for the block fp8 config |
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com> Signed-off-by: Isotr0py <2037008807@qq.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Divakar Verma <divakar.verma@amd.com> Signed-off-by: Linkun Chen <github@lkchen.net>
Signed-off-by: Divakar Verma <divakar.verma@amd.com> Signed-off-by: saeediy <saidakbarp@gmail.com>
arg.trust_remote_code
for better user exeperience. Use--trut_remote_code
when tuning for deepseekCurrently, DeepSeek-V3 runs into error when using graph-mode. This has been handled by forcing 'eager-mode' as a workaround. Hence, bypassing graph capture in the moe tuning script for deepseek-v3- use a boolean variableuse_graphs
to selectively diasble graph capture for DeepSeek-V3-get_default_config
was missingis_marlin
flag. Set it to False as default.