Skip to content

Commit 93b6805

Browse files
leslie-fang25dominicshanshan
authored andcommitted
[https://nvbugs/5652552][fix] Log the llm args (NVIDIA#9119)
Signed-off-by: leslie-fang25 <[email protected]> Signed-off-by: Mike Iovine <[email protected]> Signed-off-by: Mike Iovine <[email protected]> Signed-off-by: Wangshanshan <[email protected]>
1 parent 6445b3e commit 93b6805

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorrt_llm/_torch/pyexecutor/py_executor_creator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -732,5 +732,7 @@ def drafting_loop_wrapper(model):
732732
if mapping.rank == 0:
733733
logger.info(f"LLM Args:\n{llm_args}")
734734

735+
logger.info(f"{llm_args}")
736+
735737
py_executor.start_worker()
736738
return py_executor

0 commit comments

Comments
 (0)