TensorRT-LLMs/tensorrt_llm/runtime
Robin Kobus 20425deb3b
[https://nvbugs/5238105] fix: ModelRunnerCpp num_return_sequences (#3951)
Signed-off-by: Robin Kobus <19427718+Funatiq@users.noreply.github.com>
2025-06-06 12:31:11 +02:00
..
memory_pools Update TensorRT-LLM (#2820) 2025-02-25 21:21:49 +08:00
processor_wrapper Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
__init__.py Update TensorRT-LLM (#2110) 2024-08-13 22:34:33 +08:00
enc_dec_model_runner.py feat: [nvbugs/5261055][nvbugs/5170160] non-invasive pipeline parallelism (#4034) 2025-05-16 04:16:53 +08:00
generation.py feat: [nvbugs/5261055][nvbugs/5170160] non-invasive pipeline parallelism (#4034) 2025-05-16 04:16:53 +08:00
kv_cache_manager.py open source 7f370deb0090d885d7518c2b146399ba3933c004 (#2273) 2024-09-30 13:51:19 +02:00
medusa_utils.py Update TensorRT-LLM (#2582) 2024-12-16 21:50:47 -08:00
model_runner_cpp.py [https://nvbugs/5238105] fix: ModelRunnerCpp num_return_sequences (#3951) 2025-06-06 12:31:11 +02:00
model_runner.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
multimodal_model_runner.py fix: Mistral Small vision encoder with BS>1 (#4713) 2025-05-28 12:49:28 +08:00
redrafter_utils.py Update TensorRT-LLM (#2582) 2024-12-16 21:50:47 -08:00
session.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00