Abhinav Goyal 312f761232 [Speculative Decoding] Fixing hidden states handling in batch expansion (#7508)		2024-08-19 17:58:14 -07:00
..
e2e	[Speculative Decoding] Fixing hidden states handling in batch expansion (#7508)	2024-08-19 17:58:14 -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	[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	[Core] draft_model_runner: Implement prepare_inputs on GPU for advance_step (#6338)	2024-07-17 14:30:28 -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	[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)	2024-07-01 00:33:05 -07:00
utils.py	[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)	2024-08-18 17:57:20 -07:00