kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
2fdfa39ea8
TensorRT-LLMs/tensorrt_llm/_torch/speculative
History
dongjiyingdjy 22ff81b047
fix：fix illeagel memory access when mtp >= 2 (#3006) 
* fix - fix illeagel memory access when mtp > 2

---------

Signed-off-by: Jiying Dong <87510204+dongjiyingdjy@users.noreply.github.com>
Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
Co-authored-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
2025-04-01 13:36:45 +08:00
..
__init__.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00
eagle3.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00
interface.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00
mtp.py fix：fix illeagel memory access when mtp >= 2 (#3006) 2025-04-01 13:36:45 +08:00
utils.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00