Cyrus Leung c889d5888b

[Doc] Explicitly state that PP isn't compatible with speculative decoding yet (#10975) ...

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>

2024-12-07 17:20:49 +00:00

..

__init__.py

[Bugfix] Add __init__.py files for vllm/core/block/ and vllm/spec_decode/ (#3798)

2024-04-02 12:35:31 -07:00

batch_expansion.py

[Bugfix][SpecDecode] apply sampling parameters to target probabilities for consistency in rejection sampling. (#10198)

2024-11-27 05:07:30 +00:00

draft_model_runner.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

interfaces.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

medusa_worker.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

metrics.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

mlp_speculator_worker.py

[Core] Logprobs support in Multi-step (#7652)

2024-08-29 19:19:08 -07:00

mqa_scorer.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00

multi_step_worker.py

[Speculative Decoding] Move indices to device before filtering output (#10850)

2024-12-03 17:01:39 +08:00

ngram_worker.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

proposer_worker_base.py

[Core] Logprobs support in Multi-step (#7652)

2024-08-29 19:19:08 -07:00

smaller_tp_proposer_worker.py

[Core] Logprobs support in Multi-step (#7652)

2024-08-29 19:19:08 -07:00

spec_decode_worker.py

[Doc] Explicitly state that PP isn't compatible with speculative decoding yet (#10975)

2024-12-07 17:20:49 +00:00

target_model_runner.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00

top1_proposer.py

[Feature] [Spec decode]: Combine chunked prefill with speculative decoding (#9291)

2024-11-07 08:15:14 -08:00

util.py

Remove hard-dependencies of Speculative decode to CUDA workers (#10587)

2024-11-26 17:57:11 -08:00