Yuanle Liu 7f5f2113c2 Support keep sampling mask (#6725) ...

* naive version

* return list(int)

* fix bug: first_token's sampling mask miss

* pre-commit

* support mtp

* pre-commit

* fix ut

* fix zmq name conflits

* fix ut

* add ut

* fix ut timeout

* optimize performance

* fix

* support top_k mask

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* update comment

* update comment

* update comment

---------

Co-authored-by: Jiaxin Sui <95567040+plusNew001@users.noreply.github.com>
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

2026-03-17 20:07:31 -07:00

..

__init__.py

…

apply_penalty_multi_scores.py

…

speculate_logprob_utils.py

…

top_k_top_p_sampling.py

Support keep sampling mask (#6725)

2026-03-17 20:07:31 -07:00