TensorRT-LLMs/tensorrt_llm/_torch/speculative
Fanrong Li e6b482ef47
fix: change the seq_lens sync copy to an async one (#3786)
---------

Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
2025-04-29 23:56:49 +08:00
..
__init__.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00
eagle3.py [fix] Fix flashinfer + speculation issues (#3686) 2025-04-28 14:34:22 -04:00
interface.py fix: change the seq_lens sync copy to an async one (#3786) 2025-04-29 23:56:49 +08:00
mtp.py feat: return logits in PyTorch flow (#3221) 2025-04-24 16:56:03 -07:00
utils.py Support CUDA graphs for EAGLE3 (#3176) 2025-04-17 04:53:50 +08:00