vllm/tests/samplers
2024-05-01 08:45:42 -04: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 [Bugfix] Fix logits processor when prompt_logprobs is not None (#3899) 2024-04-10 00:09:36 -07:00
test_logprobs.py [Core] Refactoring sampler and support prompt logprob for chunked prefill (#4309) 2024-04-26 13:02:02 +00:00
test_ranks.py hotfix isort on logprobs ranks pr (#3622) 2024-03-25 11:55:46 -07:00
test_rejection_sampler.py [Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951) 2024-04-23 08:02:36 +00:00
test_sampler.py [BugFix] Fix min_tokens when eos_token_id is None (#4389) 2024-04-27 09:52:46 -07:00
test_seeded_generate.py [Test] Make model tests run again and remove --forked from pytest (#3631) 2024-03-28 21:06:40 -07:00