vllm/tests/samplers
Cyrus Leung 5ae5ed1e60
[Core] Consolidate prompt arguments to LLM engines (#4328)
Co-authored-by: Roger Wang <ywang@roblox.com>
2024-05-28 13:29:31 -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 [Test] Make model tests run again and remove --forked from pytest (#3631) 2024-03-28 21:06:40 -07:00
test_ignore_eos.py [Test] Add ignore_eos test (#4519) 2024-05-01 08:45:42 -04:00
test_logits_processor.py [Core] Consolidate prompt arguments to LLM engines (#4328) 2024-05-28 13:29:31 -07:00
test_logprobs.py [CI/Build] Move test_utils.py to tests/utils.py (#4425) 2024-05-13 23:50:09 +09:00
test_ranks.py hotfix isort on logprobs ranks pr (#3622) 2024-03-25 11:55:46 -07:00
test_rejection_sampler.py [Dynamic Spec Decoding] Auto-disable by the running queue size (#4592) 2024-05-08 21:44:00 +00:00
test_sampler.py [Misc] Remove unnecessary ModelRunner imports (#4703) 2024-05-09 00:17:17 -07:00
test_seeded_generate.py [Core] Consolidate prompt arguments to LLM engines (#4328) 2024-05-28 13:29:31 -07:00