TensorRT-LLMs/tests/unittest/_torch/sampler
Daniel Cámpora 099f081e03
[TRTLLM-7155][feat] Unify sampler handle logits implementation. (#6867)
Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>
2025-08-22 08:09:30 +02:00
..
test_beam_search.py [None][ci] move unittests to sub-directories (#6635) 2025-08-20 05:42:22 -04:00
test_best_of_n.py [None][ci] move unittests to sub-directories (#6635) 2025-08-20 05:42:22 -04:00
test_return_logits.py [TRTLLM-7155][feat] Unify sampler handle logits implementation. (#6867) 2025-08-22 08:09:30 +02:00
test_trtllm_sampler.py [None][ci] move unittests to sub-directories (#6635) 2025-08-20 05:42:22 -04:00