Travis Johnson 01b6f9e1f0 [Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047) ...

Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>

2024-09-24 17:29:56 -07:00

..

__init__.py

…

conftest.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_compatibility.py

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

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

test_eagle_correctness.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_integration_dist_tp2.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -07:00

test_integration_dist_tp4.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -07:00

test_integration.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -07:00

test_logprobs.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_medusa_correctness.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_mlp_correctness.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_multistep_correctness.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -07:00

test_ngram_correctness.py

[Core][Bugfix] Support prompt_logprobs returned with speculative decoding (#8047)

2024-09-24 17:29:56 -07:00

test_seed.py

[Speculative Decoding] Test refactor (#8317)

2024-09-11 14:07:34 -07:00