Skip to content

Commit d90d8c6

Browse files
committed
fix pre-commit hook
Signed-off-by: richardhuo-nv <[email protected]>
1 parent 2f99b38 commit d90d8c6

File tree

3 files changed

+3
-5
lines changed

3 files changed

+3
-5
lines changed

tensorrt_llm/executor/executor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,7 @@
2121
from ..bindings import executor as tllm
2222
from ..builder import Engine
2323
from ..disaggregated_params import DisaggregatedParams
24-
from ..llmapi.llm_args import KvCacheConnectorConfig
25-
from ..llmapi.llm_args import TorchLlmArgs
24+
from ..llmapi.llm_args import KvCacheConnectorConfig, TorchLlmArgs
2625
from ..llmapi.llm_utils import KvCacheRetentionConfig
2726
from ..llmapi.mpi_session import (MpiSession, external_mpi_comm_available,
2827
need_spawn_mpi_workers)

tensorrt_llm/executor/worker.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,7 @@
1818
mpi_comm, mpi_rank, nvtx_range_debug)
1919
from ..bindings import executor as tllm
2020
from ..builder import ConfigEncoder, Engine, EngineConfig
21-
from ..llmapi.llm_args import KvCacheConnectorConfig, PybindMirror
22-
from ..llmapi.llm_args import PybindMirror, TorchLlmArgs
21+
from ..llmapi.llm_args import KvCacheConnectorConfig, PybindMirror, TorchLlmArgs
2322
from ..llmapi.mpi_session import set_mpi_session_cpp
2423
from ..llmapi.tokenizer import TokenizerBase
2524
from ..llmapi.tracer import VizTracer, global_tracer, set_global_tracer

tensorrt_llm/llmapi/llm_args.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2321,7 +2321,7 @@ class TorchLlmArgs(BaseLlmArgs):
23212321
default=None,
23222322
description="The config for KV cache connector.",
23232323
)
2324-
2324+
23252325
mm_encoder_only: bool = Field(
23262326
default=False,
23272327
description=

0 commit comments

Comments
 (0)