diff --git a/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py b/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py index bcf18ffa2a2..a8ad9cd4b35 100644 --- a/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py +++ b/tensorrt_llm/_torch/pyexecutor/py_executor_creator.py @@ -686,7 +686,8 @@ def drafting_loop_wrapper(model): _adjust_torch_mem_fraction(pytorch_backend_config) - logger.info(f"{llm_args}") + if mapping.rank == 0: + logger.info(f"LLM Args:\n{llm_args}") py_executor.start_worker() return py_executor