William Lin 57b7be0e1c [Speculative decoding] [Multi-Step] decouple should_modify_greedy_probs_inplace (#6971)

2024-08-09 05:42:45 +00: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 (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_ignore_eos.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_logits_processor.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_logprobs.py

[Core] Reduce unnecessary compute when logprobs=None (#6532)

2024-07-29 16:47:31 +00:00

test_ranks.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_rejection_sampler.py

[SpecDecode] [Minor] Fix spec decode sampler tests (#7183)

2024-08-06 10:40:32 -07:00

test_sampler.py

[Speculative decoding] [Multi-Step] decouple should_modify_greedy_probs_inplace (#6971)

2024-08-09 05:42:45 +00:00

test_seeded_generate.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_typical_acceptance_sampler.py

[SpecDecode] [Minor] Fix spec decode sampler tests (#7183)

2024-08-06 10:40:32 -07:00