vllm/tests/samplers
Travis Johnson c13ad1b7bd
feat: implement the min_tokens sampling parameter (#3124)
Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>
Co-authored-by: Nick Hill <nickhill@us.ibm.com>
2024-03-25 10:14:26 -07:00
..
test_beam_search.py [Core] Refactor Attention Take 2 (#3462) 2024-03-25 04:39:33 +00:00
test_logprobs.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_ranks.py [Core] Adding token ranks along with logprobs (#3516) 2024-03-25 10:13:10 -07:00
test_rejection_sampler.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_sampler.py feat: implement the min_tokens sampling parameter (#3124) 2024-03-25 10:14:26 -07:00
test_seeded_generate.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00