Erin 83f37614ef feat: Support Top-K logprobs and prompt_logprobs in LLMAPI (#3388) ... * support return logprob in llmapi Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com> update and add test Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com> stability test Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com> * revert removal of old flag Signed-off-by: Erin Ho <erinh@nvidia.com> Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com> --------- Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com> Signed-off-by: Erin Ho <erinh@nvidia.com>		2025-05-01 12:47:14 -04:00
..
__init__.py	cacheTransceiver buffer manager (#3798)	2025-04-27 11:48:15 +08:00
_perf_evaluator.py	Add thread leak check and fix thread/memory leak issues. (#3270)	2025-04-08 19:03:18 +08:00
build_cache.py	Update TensorRT-LLM (#2413)	2024-11-05 16:27:06 +08:00
disagg_utils.py	feat: Disaggregated router class (#3584)	2025-04-19 00:34:12 +08:00
llm_args.py	feat: add relaxed acceptance for DS (#3865)	2025-05-01 21:50:36 +08:00
llm_utils.py	fix: trtllm-bench build trt engine on slurm (#3825)	2025-04-27 22:26:23 +08:00
llm.py	feat: Support Top-K logprobs and prompt_logprobs in LLMAPI (#3388)	2025-05-01 12:47:14 -04:00
mgmn_leader_node.py	fix: trtllm-bench build trt engine on slurm (#3825)	2025-04-27 22:26:23 +08:00
mgmn_worker_node.py	Update TensorRT-LLM (#2333)	2024-10-15 15:28:40 +08:00
mpi_session.py	fix: trtllm-bench build trt engine on slurm (#3825)	2025-04-27 22:26:23 +08:00
tokenizer.py	test [TRTLLM-4477,TRTLLM-4481]: Accuracy test improvement (Part 3.5): Support GSM8K and GPQA (#3483)	2025-04-22 07:38:16 +08:00
tracer.py	Update TensorRT-LLM (#2413)	2024-11-05 16:27:06 +08:00
trtllm-llmapi-launch	fix hmac in remote mpi session (#3649)	2025-04-18 17:47:51 +08:00
utils.py	fix: trtllm-bench build trt engine on slurm (#3825)	2025-04-27 22:26:23 +08:00