Cyrus Leung 59a0192fb9

[Core] Interface for accessing model from VllmRunner (#10353) ...

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

2025-01-20 15:00:59 +08: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

[core] platform agnostic executor via collective_rpc (#11256)

2025-01-15 13:45:21 +08:00

metrics.py

[platforms] enable platform plugins (#11602)

2024-12-30 20:24:45 +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

[core] platform agnostic executor via collective_rpc (#11256)

2025-01-15 13:45:21 +08:00

ngram_worker.py

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +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] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08:00

spec_decode_worker.py

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08: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