Woo-Yeon Lee 2ce5d6688b

[Speculative Decoding] Support draft model on different tensor-parallel size than target model (#5414)

2024-06-25 09:56:06 +00:00

..

__init__.py

[Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951)

2024-04-23 08:02:36 +00:00

conftest.py

[Bugfix] AsyncLLMEngine hangs with asyncio.run (#5654)

2024-06-19 13:57:12 -07:00

test_compatibility.py

[Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840)

2024-05-16 00:53:51 -07:00

test_integration_dist_tp2.py

[Speculative Decoding] Support draft model on different tensor-parallel size than target model (#5414)

2024-06-25 09:56:06 +00:00

test_integration_dist_tp4.py

[Speculative Decoding] Support draft model on different tensor-parallel size than target model (#5414)

2024-06-25 09:56:06 +00:00

test_integration.py

[Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840)

2024-05-16 00:53:51 -07:00

test_logprobs.py

[Speculative decoding] Support target-model logprobs (#4378)

2024-05-03 15:52:01 -07:00

test_multistep_correctness.py

[Speculative decoding][Re-take] Enable TP>1 speculative decoding (#4840)

2024-05-16 00:53:51 -07:00

test_ngram_correctness.py

[Dynamic Spec Decoding] Minor fix for disabling speculative decoding (#5000)

2024-05-25 10:00:14 -07:00