Skip to content

Commit 702b5ac

Browse files
jayfeather9yujinbiao
authored andcommitted
add cli param fix
1 parent ee103a3 commit 702b5ac

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lightllm/server/api_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def make_argument_parser() -> argparse.ArgumentParser:
312312
"--visual_infer_batch_size", type=int, default=1, help="number of images to process in each inference batch"
313313
)
314314
parser.add_argument(
315-
"--visual_gpu_ids", nargs="+", type=int, default=None, help="List of GPU IDs to use, e.g., 0 1 2"
315+
"--visual_gpu_ids", nargs="+", type=int, default=[0, 1, 2, 3, 4, 5, 6, 7], help="List of GPU IDs to use, e.g., 0 1 2"
316316
)
317317
parser.add_argument("--visual_tp", type=int, default=1, help="number of tensort parallel instances for ViT")
318318
parser.add_argument("--visual_dp", type=int, default=1, help="number of data parallel instances for ViT")

lightllm/server/api_start.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def normal_or_p_d_start(args):
154154

155155
# if use_hi_dynamic_prompt_cache, then use_dynamic_prompt_cache must be True
156156
if args.use_hi_dynamic_prompt_cache:
157-
assert args.use_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
157+
assert not args.disable_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
158158

159159
# help to manage data stored on Ceph
160160
if "s3://" in args.model_dir:

0 commit comments

Comments
 (0)