TensorRT-LLMs/tensorrt_llm/_torch/speculative
Izzy Putterman b36460d7b5
[None][feat] Deepseek: Start Eagle work (#6210)
Signed-off-by: Izzy Putterman <iputterman@nvidia.com>
Co-authored-by: Mike Iovine <miovine@nvidia.com>
2025-08-22 12:57:17 -04:00
..
__init__.py [None][feat] Clean up ngram auto mode, add max_concurrency to configs (#6676) 2025-08-07 12:51:47 -04:00
auto_heuristic.py [None][feat] Clean up ngram auto mode, add max_concurrency to configs (#6676) 2025-08-07 12:51:47 -04:00
drafter.py [None][feat] Optimize CUDA graph memory usage for spec decode cases (#6718) 2025-08-08 13:56:53 -04:00
eagle3.py [None][feat] Deepseek: Start Eagle work (#6210) 2025-08-22 12:57:17 -04:00
interface.py [None][feat] Deepseek: Start Eagle work (#6210) 2025-08-22 12:57:17 -04:00
model_drafter.py [TRTLLM-7155][feat] Unify sampler handle logits implementation. (#6867) 2025-08-22 08:09:30 +02:00
mtp.py [TRTLLM-7155][feat] Unify sampler handle logits implementation. (#6867) 2025-08-22 08:09:30 +02:00
ngram.py [https://nvbugs/5452167][fix] Fix ngram padding issue (#6837) 2025-08-13 11:23:16 +08:00
utils.py [None][feat] Deepseek: Start Eagle work (#6210) 2025-08-22 12:57:17 -04:00