Yuxian Qiu c8e062bfd3 fix: [nvbugs/5287097] Align PP layer distribution between pytorch and TRT flow. (#4399) ... Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> Signed-off-by: Aurelien Chartier <2567591+achartier@users.noreply.github.com> Co-authored-by: Aurelien Chartier <2567591+achartier@users.noreply.github.com>		2025-05-19 14:25:36 -07: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	fix: [nvbugs/5287097] Align PP layer distribution between pytorch and TRT flow. (#4399)	2025-05-19 14:25:36 -07:00
model_runner.py	Update TensorRT-LLM (#2755)	2025-02-11 03:01:00 +00:00
multimodal_model_runner.py	feat: Support Mistral Small 3.1 24B VLM in TRT workflow (#4183)	2025-05-14 03:47:22 +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