kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-04 02:02:01 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
cbca6505ff
TensorRT-LLMs/tests/unittest/_torch/speculative
History
Fanrong Li 77f8e43592
[fix] Fix relaxed acceptance to support enabling it in context phase (#4126) 
* fix relaxed acceptance to support enable this feature in context phase.

Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>

* fix sample_and_accept_draft_tokens unit test.

Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>

---------

Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
2025-05-09 14:11:14 +08:00
..
test_eagle3.py [fix] Fix llama4 + eagle3 (#3998) 2025-05-08 19:20:27 -04:00
test_mtp_prepare_drafter_inputs.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_mtp_sample_and_accept_draft_tokens.py [fix] Fix relaxed acceptance to support enabling it in context phase (#4126) 2025-05-09 14:11:14 +08:00
test_mtp_update_mtp_hidden_states.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00