TensorRT-LLMs/tests/unittest/_torch/speculative
Izzy Putterman e607768e45
Speculation: Draft Target in new FW (#4558)
Signed-off-by: Izzy Putterman <iputterman@nvidia.com>
2025-06-17 02:26:08 +08:00
..
test_draft_target.py Speculation: Draft Target in new FW (#4558) 2025-06-17 02:26:08 +08:00
test_eagle3.py Speculation: Draft Target in new FW (#4558) 2025-06-17 02:26:08 +08: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
test_ngram.py ci: [nvbugs/5280806] Unwaive unittests/_torch. (#4951) 2025-06-09 19:04:11 +08:00