Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions tensorrt_llm/llmapi/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -1007,6 +1007,9 @@ def _build_model(self):

# TODO: revisit gather_context_logits
return_logits = self.args.gather_generation_logits

logger.info(f"{self.args}")
Comment thread
leslie-fang25 marked this conversation as resolved.
Outdated

self._executor = self._executor_cls.create(
self._engine_dir,
executor_config=None,
Expand Down
Loading