TensorRT-LLMs/tensorrt_llm/_torch/speculative
Gaoji Liu 9d2df04a72
[None][doc] fix mtp.py typo (#9307)
Signed-off-by: liugaoji <757394026@qq.com>
2025-11-30 21:55:13 -08:00
..
__init__.py [None][feat] Draft: Save state first pass (#7012) 2025-10-01 18:40:55 -04:00
auto_heuristic.py [None][feat] Clean up ngram auto mode, add max_concurrency to configs (#6676) 2025-08-07 12:51:47 -04:00
drafter.py [TRTLLM-8136][feat] Dynamic draft length in spec decode (stage 1). (#8194) 2025-11-18 11:13:39 -05:00
drafting_loops.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
eagle3.py [None][feat] Add environment variable to force spec-dec number of accepted tokens (#9371) 2025-11-26 07:22:16 -08:00
interface.py [None][feat] Add environment variable to force spec-dec number of accepted tokens (#9371) 2025-11-26 07:22:16 -08:00
model_drafter.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
mtp.py [None][doc] fix mtp.py typo (#9307) 2025-11-30 21:55:13 -08:00
ngram.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
save_hidden_state.py [TRTLLM-8160][feat] Add max_total_draft_tokens (#8366) 2025-10-21 11:11:04 -04:00
spec_tree_manager.py [TRTLLM-8160][feat] Add draft token tree runtime on CDL (#8586) 2025-11-25 09:40:55 -05:00
speculation_gate.py [TRTLLM-7412][feat] Turn off spec decode when the rolling average acceptance length drops below threshold. (#7283) 2025-10-13 15:51:14 -07:00
utils.py [TRTLLM-7954][feat] Target model KV cache rellocation (#8421) 2025-10-23 09:36:50 +08:00