Sirej Dua 15aba081f3 [Speculative Decoding] MLPSpeculator Tensor Parallel support (1/2) (#6050) ...

Co-authored-by: Sirej Dua <sirej.dua@databricks.com>
Co-authored-by: Sirej Dua <Sirej Dua>

2024-07-02 07:20:29 -07:00

..

e2e

[Speculative Decoding] MLPSpeculator Tensor Parallel support (1/2) (#6050)

2024-07-02 07:20:29 -07:00

__init__.py

[Speculative decoding 3/9] Worker which speculates, scores, and applies rejection sampling (#3103)

2024-03-08 23:32:46 -08:00

test_batch_expansion.py

[Core] Optimize block_manager_v2 vs block_manager_v1 (to make V2 default) (#5602)

2024-07-01 20:10:37 -07:00

test_dynamic_spec_decode.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

test_metrics.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

test_multi_step_worker.py

[Spec Decode] Introduce DraftModelRunner (#5799)

2024-06-28 09:17:51 -07:00

test_ngram_worker.py

[Speculative Decoding] Add ProposerWorkerBase abstract class (#5252)

2024-06-05 14:53:05 -07:00

test_spec_decode_worker.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

test_utils.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

utils.py

[Spec Decode] Introduce DraftModelRunner (#5799)

2024-06-28 09:17:51 -07:00