TensorRT-LLMs/tensorrt_llm/_torch/speculative
Yuan Tong d4c0423cdb
refactor: collect executor and decoder states into dataclass (#3234)
* fix: Proper error bubbling for PyExecutor

Signed-off-by: Yuan Tong <13075180+tongyuantongyu@users.noreply.github.com>
Co-authored-by: QI JUN <22017000+QiJune@users.noreply.github.com>
2025-04-15 16:31:45 +08:00
..
__init__.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00
eagle3.py refactor: collect executor and decoder states into dataclass (#3234) 2025-04-15 16:31:45 +08:00
interface.py refactor: collect executor and decoder states into dataclass (#3234) 2025-04-15 16:31:45 +08:00
mtp.py refactor: collect executor and decoder states into dataclass (#3234) 2025-04-15 16:31:45 +08:00
utils.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00