Skip to content

Commit d009638

Browse files
authored
add cli param fix
1 parent d26cabd commit d009638

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
@@ -304,7 +304,7 @@ def make_argument_parser() -> argparse.ArgumentParser:
304304
"--visual_infer_batch_size", type=int, default=1, help="number of images to process in each inference batch"
305305
)
306306
parser.add_argument(
307-
"--visual_gpu_ids", nargs="+", type=int, default=[0], help="List of GPU IDs to use, e.g., 0 1 2"
307+
"--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"
308308
)
309309
parser.add_argument("--visual_tp", type=int, default=1, help="number of tensort parallel instances for ViT")
310310
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
@@ -148,7 +148,7 @@ def normal_or_p_d_start(args):
148148

149149
# if use_hi_dynamic_prompt_cache, then use_dynamic_prompt_cache must be True
150150
if args.use_hi_dynamic_prompt_cache:
151-
assert args.use_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
151+
assert not args.disable_dynamic_prompt_cache, "use_hi_dynamic_prompt_cache must be used with use_dynamic_prompt_cache"
152152

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

0 commit comments

Comments
 (0)