Skip to content

Commit 73f01a9

Browse files
authored
fix cli serve --help (#3895)
1 parent f710f51 commit 73f01a9

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

lmdeploy/cli/utils.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -572,7 +572,7 @@ def enable_eplb(parser):
572572
@staticmethod
573573
def enable_metrics(parser):
574574
"""Add argument enable_metrics to parser."""
575-
parser.add_argument('--enable-metrics', action='store_true', default=False, help='enable metrics system')
575+
return parser.add_argument('--enable-metrics', action='store_true', default=False, help='enable metrics system')
576576

577577
# For Disaggregation
578578
@staticmethod
@@ -596,19 +596,19 @@ def migration_backend(parser):
596596
@staticmethod
597597
def disable_vision_encoder(parser):
598598
"""Disable loading vision encoder."""
599-
parser.add_argument('--disable-vision-encoder',
600-
action='store_true',
601-
default=False,
602-
help='enable metrics system')
599+
return parser.add_argument('--disable-vision-encoder',
600+
action='store_true',
601+
default=False,
602+
help='enable metrics system')
603603

604604
@staticmethod
605605
def logprobs_mode(parser):
606606
"""The mode of logprobs."""
607-
parser.add_argument('--logprobs-mode',
608-
type=str,
609-
default=None,
610-
choices=[None, 'raw_logits', 'raw_logprobs'],
611-
help='The mode of logprobs.')
607+
return parser.add_argument('--logprobs-mode',
608+
type=str,
609+
default=None,
610+
choices=[None, 'raw_logits', 'raw_logprobs'],
611+
help='The mode of logprobs.')
612612

613613

614614
# adapted from https://github.com/vllm-project/vllm/blob/main/vllm/utils/__init__.py

0 commit comments

Comments
 (0)