[misc] Use --cuda-graph-max-bs-decode in tests, examples, and docs (#29591)
This commit is contained in:
@@ -211,7 +211,7 @@ def init_process_dst(
|
||||
str(base_gpu_id),
|
||||
"--tp-size",
|
||||
str(tp_size),
|
||||
"--cuda-graph-max-bs",
|
||||
"--cuda-graph-max-bs-decode",
|
||||
2,
|
||||
"--tokenizer-path",
|
||||
model_name,
|
||||
|
||||
@@ -222,7 +222,7 @@ def init_process_dst(
|
||||
str(base_gpu_id),
|
||||
"--tp-size",
|
||||
str(tp_size),
|
||||
"--cuda-graph-max-bs",
|
||||
"--cuda-graph-max-bs-decode",
|
||||
2,
|
||||
"--tokenizer-path",
|
||||
model_name,
|
||||
|
||||
Reference in New Issue
Block a user