Skip to content

Spec dec#37

Merged
HaibaraAiChan merged 2 commits into
mainfrom
spec_dec
Jan 14, 2026
Merged

Spec dec#37
HaibaraAiChan merged 2 commits into
mainfrom
spec_dec

Conversation

@xiongxu1998
Copy link
Copy Markdown
Collaborator

  1. add tree-based speculative decoding
  2. refactoring the architecture of SD to make it more clear
  3. add simple(Top K) and adaptive(MLP) pruner to reduce the transformation data between multiple devices
  4. currently, batch inference for SD is not support as it's so complicated that I still need some time to debug it.

@HaibaraAiChan HaibaraAiChan merged commit 68a4c08 into main Jan 14, 2026
1 check passed
@HaibaraAiChan HaibaraAiChan deleted the spec_dec branch January 14, 2026 01:25
JiuChen0 pushed a commit to JiuChen0/BloomBee that referenced this pull request Mar 22, 2026
* 重构pruner代码

* add adaptive pruner

---------

Co-authored-by: Xu Xiong <xiongxu1998@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants