Skip to content

Commit 39ca128

Browse files
jayfeather9jinbiaoyu
authored andcommitted
add cli param fix
1 parent 8fd842c commit 39ca128

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
@@ -160,7 +160,7 @@ def normal_or_p_d_start(args):
160160

161161
# if use_hi_dynamic_prompt_cache, then use_dynamic_prompt_cache must be True
162162
if args.use_hi_dynamic_prompt_cache:
163-
assert args.use_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
163+
assert not args.disable_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
164164

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

0 commit comments

Comments
 (0)