vllm/tests/samplers
Thomas Parnell d4201e06d5
[Bugfix] Make spec. decode respect per-request seed. (#6034)
Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
Co-authored-by: Nick Hill <nickhill@us.ibm.com>
2024-07-18 19:22:08 -07: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 [CI/Test] improve robustness of test (vllm_runner) (#5357) 2024-06-08 08:59:20 +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 [Bugfix] use float32 precision in samplers/test_logprobs.py for comparing with HF (#6409) 2024-07-15 13:14:49 -04: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 [Bugfix] Make spec. decode respect per-request seed. (#6034) 2024-07-18 19:22:08 -07:00
test_sampler.py [CI/Build] Fix Args for _get_logits_warper in Sampler Test (#5922) 2024-06-27 11:43:04 -07: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 [Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348) 2024-07-01 00:33:05 -07:00