[TRTLLM-14865][feat] Support occurrence penalties with beam search for TorchSampler - #17189

Merged

lori-ren merged 7 commits into

NVIDIA:mainfrom

lori-ren:feat/beam-search-penalties

Aug 13, 2026

Commits on Aug 13, 2026

[TRTLLM-14865][feat] support occurrence penalties with beam search
lori-ren
committed
5c248cc
[TRTLLM-14865][feat] stage beam widths and merge the beam counts update
lori-ren
committed
869511d
[TRTLLM-14865][test] Move the beam-search penalty e2e case to test_penalties_e2e.py
lori-ren
committed
9139366
[TRTLLM-14865][chore] Require max_beam_width and document the per-beam history
lori-ren
committed
6c9e06e
Merge branch 'main' into feat/beam-search-penalties
lori-ren
authored
8c1bcdf
[TRTLLM-14865][fix] Drop two redundant casts flagged by mypy in sampler_strategy.py
lori-ren
committed
0c19520
[TRTLLM-14865][fix] Restate the beam-search narrowing for the full mypy check
lori-ren
committed
41cf62e