Skip to content

Commit b41071f

Browse files
committed
lint
Signed-off-by: Suyog Gupta <[email protected]>
1 parent b3955bc commit b41071f

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

docs/source/torch/auto_deploy/advanced/serving_with_trtllm_serve.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,4 +75,3 @@ attn_backend: flashinfer # recommended for best performance
7575

7676
- [AutoDeploy overview](../auto-deploy.md)
7777
- [Benchmarking with trtllm-bench](./benchmarking_with_trtllm_bench.md)
78-

tensorrt_llm/commands/serve.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212
from torch.cuda import device_count
1313

1414
from tensorrt_llm import LLM as PyTorchLLM
15-
from tensorrt_llm._torch.auto_deploy.llm import LLM as AutoDeployLLM
1615
from tensorrt_llm import MultimodalEncoder
1716
from tensorrt_llm._tensorrt_engine import LLM
17+
from tensorrt_llm._torch.auto_deploy.llm import LLM as AutoDeployLLM
1818
from tensorrt_llm._utils import mpi_rank
1919
from tensorrt_llm.executor.utils import LlmLauncherEnvs
2020
from tensorrt_llm.llmapi import (BuildConfig, CapacitySchedulerPolicy,
@@ -211,10 +211,13 @@ def launch_mm_encoder_server(
211211
default="localhost",
212212
help="Hostname of the server.")
213213
@click.option("--port", type=int, default=8000, help="Port of the server.")
214-
@click.option("--backend",
215-
type=click.Choice(["pytorch", "trt", "_autodeploy"]),
216-
default="pytorch",
217-
help="Set to 'pytorch' for pytorch path and '_autodeploy' for autodeploy path. Default is pytorch path.")
214+
@click.option(
215+
"--backend",
216+
type=click.Choice(["pytorch", "trt", "_autodeploy"]),
217+
default="pytorch",
218+
help=
219+
"Set to 'pytorch' for pytorch path and '_autodeploy' for autodeploy path. Default is pytorch path."
220+
)
218221
@click.option('--log_level',
219222
type=click.Choice(severity_map.keys()),
220223
default='info',

0 commit comments

Comments
 (0)