vllm/tests/samplers
youkaichao 9fb900f90c
[CI/Test] improve robustness of test (hf_runner) (#5347)
[CI/Test] improve robustness of test by replacing del with context manager (hf_runner) (#5347)
2024-06-07 22:31:32 -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 (hf_runner) (#5347) 2024-06-07 22:31:32 -07:00
test_ignore_eos.py Update test_ignore_eos (#4898) 2024-06-02 02:21:53 +00: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/Test] improve robustness of test (hf_runner) (#5347) 2024-06-07 22:31:32 -07: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