vllm/tests/samplers
Vasiliy Alekseev 07e981fdf4
[Frontend] Bad words sampling parameter (#9717)
Signed-off-by: Vasily Alexeev <alvasian@yandex.ru>
2024-10-26 16:29:38 +00:00
..
__init__.py [CI/Build] Move test_utils.py to tests/utils.py (#4425) 2024-05-13 23:50:09 +09:00
test_beam_search.py [core] remove beam search from the core (#9105) 2024-10-07 05:47:04 +00:00
test_ignore_eos.py [CI/Test] improve robustness of test (vllm_runner) (#5357) 2024-06-08 08:59:20 +00:00
test_logits_processor.py [CI/Test] improve robustness of test (vllm_runner) (#5357) 2024-06-08 08:59:20 +00:00
test_logprobs.py [Core] Reduce unnecessary compute when logprobs=None (#6532) 2024-07-29 16:47:31 +00:00
test_no_bad_words.py [Frontend] Bad words sampling parameter (#9717) 2024-10-26 16:29:38 +00:00
test_ranks.py [CI/Test] improve robustness of test (vllm_runner) (#5357) 2024-06-08 08:59:20 +00:00
test_rejection_sampler.py [SpecDec][Misc] Cleanup, remove bonus token logic. (#8701) 2024-09-22 12:34:14 -07:00
test_sampler.py [core] remove beam search from the core (#9105) 2024-10-07 05:47:04 +00:00
test_seeded_generate.py [CI/Test] improve robustness of test (vllm_runner) (#5357) 2024-06-08 08:59:20 +00:00
test_typical_acceptance_sampler.py Fix typical acceptance sampler with correct recovered token ids (#8562) 2024-09-23 12:32:27 -07:00