Cade DanielandGitHub 79d64c4954 [Speculative decoding 1/9] Optimized rejection sampler (#2336)

2024-01-09 15:38:41 -08:00

..

test_beam_search.py

Align vLLM's beam search implementation with HF generate (#857)

2023-09-04 17:29:42 -07:00

test_logprobs.py

Implement prompt logprobs & Batched topk for computing logprobs (#1328)

2023-10-16 10:56:50 -07:00

test_rejection_sampler.py

[Speculative decoding 1/9] Optimized rejection sampler (#2336)

2024-01-09 15:38:41 -08:00

test_sampler.py

Fix broken sampler tests (#1896)

2023-12-02 16:06:17 -08:00