vllm/tests/spec_decode
Cody Yu 973617ae02
[Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840)
Co-authored-by: Cade Daniel <edacih@gmail.com>
Co-authored-by: Cade Daniel <cade@anyscale.com>
2024-05-16 00:53:51 -07:00
..
e2e [Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840) 2024-05-16 00:53:51 -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 [Misc] Add pytest marker to opt-out of global test cleanup (#3863) 2024-04-04 21:54:16 -07:00
test_dynamic_spec_decode.py [Dynamic Spec Decoding] Auto-disable by the running queue size (#4592) 2024-05-08 21:44:00 +00:00
test_metrics.py [Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951) 2024-04-23 08:02:36 +00:00
test_multi_step_worker.py [Misc][Refactor] Introduce ExecuteModelData (#4540) 2024-05-03 17:47:07 -07:00
test_ngram_worker.py [Speculative decoding] Improve n-gram efficiency (#4724) 2024-05-13 15:00:13 -07:00
test_spec_decode_worker.py [Misc][Refactor] Introduce ExecuteModelData (#4540) 2024-05-03 17:47:07 -07:00
test_utils.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
utils.py [Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734) 2024-05-11 11:30:37 -07:00