feat(pytorch): support Kimi K2.6 - #4846
Open
qescccczmr wants to merge 8 commits into
Open
Conversation
qescccczmr
force-pushed
the
feat/kimi-k2.6-pytorch
branch
4 times, most recently
from
August 17, 2026 11:13
dd00426 to
707e933
Compare
qescccczmr
force-pushed
the
feat/kimi-k2.6-pytorch
branch
from
August 17, 2026 13:15
707e933 to
fb64e19
Compare
qescccczmr
force-pushed
the
feat/kimi-k2.6-pytorch
branch
3 times, most recently
from
August 19, 2026 11:34
fb64e19 to
33315ea
Compare
qescccczmr
marked this pull request as ready for review
August 19, 2026 14:11
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
grimoire
reviewed
Aug 20, 2026
| # DeepseekV2ModelConfigBuilder also recognizes ``kimi_k2``. Register the | ||
| # narrower Kimi builder immediately before it without changing the shared | ||
| # DeepSeek builder. | ||
| _builders = AutoModelConfigBuilder._sub_classes |
Collaborator
There was a problem hiding this comment.
You don't have to patch the builder of dsv2. You can just remove 'kimi_k2' in it's condition.
Collaborator
Author
There was a problem hiding this comment.
I've removed kimi_k2 from DeepseekV2ModelConfigBuilder.
Collaborator
|
Please fix lint |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
Add Kimi-K2.6 support to the PyTorch backend, including text and multimodal inference, compressed-tensors W4A16 MoE, EAGLE3 speculative decoding, and serving integration.
The implementation supports Kimi-K2.6's MLA attention, MoE routing, vision encoder, multimodal projector, and external EAGLE3 draft model.
Changes
Acc Benchmarks
Academic
LMDeploy and Hugging Face were evaluated with TP8 under aligned evaluation settings.
LMDeploy reaches or slightly exceeds the Hugging Face reference on both academic benchmarks, with no observed text accuracy regression.
Multimodal
Kimi-K2.6 successfully completes the MathVision multimodal evaluation.
Acceptance of EAGLE3
Draft model:
lightseekorg/kimi-k2.6-eagle3.1-mla
Aggregate results
Per-position acceptance rates:
Runtime Validation
Assistance
Assisted with Codex, reviewed manually.