TensorRT-LLMs/tensorrt_llm/_torch/speculative
Fanrong Li 1fe64b90be
fix: fix the acceptance rate of pytorch workflow in trtllm-bench (#3240)
* fix acceptance rate of pytorch workflow.
* revert the RequestOutput API change.

---------

Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
2025-04-03 15:12:24 +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 the acceptance rate of pytorch workflow in trtllm-bench (#3240) 2025-04-03 15:12:24 +08:00
utils.py Add initial EAGLE-3 implementation (#3035) 2025-03-29 22:31:24 +08:00