TensorRT-LLMs/tensorrt_llm/_torch/speculative
wili 8ecdeee300
[refactor] Simplification of Speculative decoding configs - Part 2 (#5936)
Signed-off-by: wili-65535 <wili-65535@users.noreply.github.com>
Co-authored-by: wili-65535 <wili-65535@users.noreply.github.com>
2025-07-23 09:20:27 +08:00
..
__init__.py [refactor] Simplification of Speculative decoding configs - Part 2 (#5936) 2025-07-23 09:20:27 +08:00
drafter.py [TRTLLM-6352][feat] Migrate EAGLE3 and draft/target speculation to Drafter (#6007) 2025-07-17 21:15:01 +08:00
eagle3.py [https://nvbugs/5355316] fix: update torch.compile option to fix triton store_cubin error (#5865) 2025-07-14 17:17:30 +08:00
interface.py [fix] Correct the returned value of has_spec_drafter (#6178) 2025-07-21 11:38:59 -04:00
model_drafter.py [refactor] Simplification of Speculative decoding configs - Part 2 (#5936) 2025-07-23 09:20:27 +08:00
mtp.py [nvbug/5393888][nvbug/5393042] Always use py_seq_slot (#6147) 2025-07-18 22:45:16 +03:00
ngram.py [TRTLLM-6352][feat] Migrate EAGLE3 and draft/target speculation to Drafter (#6007) 2025-07-17 21:15:01 +08:00
utils.py [refactor] Simplification of Speculative decoding configs - Part 2 (#5936) 2025-07-23 09:20:27 +08:00