Nick Hill 1856aff4d6

[Spec Decoding] Streamline batch expansion tensor manipulation (#7851)

2024-08-25 15:45:14 -07:00

..

e2e

[Bugfix] spec decode handle None entries in topk args in create_sequence_group_output (#7232)

2024-08-22 09:33:48 -04: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

[BugFix] Fix use of per-request seed with pipeline parallel (#6698)

2024-07-30 10:40:08 -07:00

test_dynamic_spec_decode.py

[Spec Decode] Disable Log Prob serialization to CPU for spec decoding for both draft and target models. (#6485)

2024-07-20 23:58:58 -07:00

test_metrics.py

[Bugfix] [SpecDecode] AsyncMetricsCollector: update time since last collection (#6578)

2024-07-19 14:01:03 -07:00

test_multi_step_worker.py

[Speculative Decoding] EAGLE Implementation with Top-1 proposer (#6830)

2024-08-22 02:42:24 -07:00

test_ngram_worker.py

[Speculative Decoding] Enabling bonus token in speculative decoding for KV cache based models (#5765)

2024-07-10 16:02:47 -07:00

test_spec_decode_worker.py

[Speculative decoding] Add periodic log with time spent in proposal/scoring/verification (#6963)

2024-08-05 08:46:44 +00:00

test_utils.py

[Spec Decoding] Streamline batch expansion tensor manipulation (#7851)

2024-08-25 15:45:14 -07:00

utils.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -07:00