Commit 249dc71
[TRTLLM-6756][feat] Update BeamSearch for TorchSampler (NVIDIA#9660)
Signed-off-by: Stefan Niebler <82932102+stnie@users.noreply.github.com>1 parent 939063a commit 249dc71
File tree
5 files changed
+329
-327
lines changed- tensorrt_llm
- _torch/pyexecutor
- tests/unittest/_torch/sampler
5 files changed
+329
-327
lines changed
0 commit comments