[https://nvbugs/5652552][fix] add printing for llm args (#9205)

Signed-off-by: Ruodi Lu <[email protected]>
Co-authored-by: Ruodi Lu <[email protected]>

Author: ruodil

tensorrt_llm/_torch/pyexecutor/py_executor_creator.py

@@ -707,7 +707,8 @@ def drafting_loop_wrapper(model):
 
     _adjust_torch_mem_fraction()
 
-    logger.info(f"{llm_args}")
+    if mapping.rank == 0:
+        logger.info(f"LLM Args:\n{llm_args}")
 
     py_executor.start_worker()
     return py_executor