vllm/tests/samplers
Antoni Baum 22de45235c
Push logprob generation to LLMEngine (#3065)
Co-authored-by: Avnish Narayan <avnish@anyscale.com>
2024-03-04 19:54:06 +00:00
..
test_beam_search.py [FIX] Fix beam search test (#2930) 2024-02-20 14:37:39 -08:00
test_logprobs.py Push logprob generation to LLMEngine (#3065) 2024-03-04 19:54:06 +00:00
test_rejection_sampler.py Remove hardcoded device="cuda" to support more devices (#2503) 2024-02-01 15:46:39 -08:00
test_sampler.py Support per-request seed (#2514) 2024-02-21 11:47:00 -08:00
test_seeded_generate.py Support per-request seed (#2514) 2024-02-21 11:47:00 -08:00