Skip to content

Commit daa31d7

Browse files
[https://nvbugs/5652552][fix] Log the llm args for main branch (#9120)
Signed-off-by: leslie-fang25 <leslief@nvidia.com>
1 parent b51258a commit daa31d7

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
@@ -707,5 +707,7 @@ def drafting_loop_wrapper(model):
707707

708708
_adjust_torch_mem_fraction()
709709

710+
logger.info(f"{llm_args}")
711+
710712
py_executor.start_worker()
711713
return py_executor

0 commit comments

Comments
 (0)