diff --git a/python/sglang/srt/model_executor/cuda_graph_runner.py b/python/sglang/srt/model_executor/cuda_graph_runner.py index 75f085ae4..bc2c4abb0 100644 --- a/python/sglang/srt/model_executor/cuda_graph_runner.py +++ b/python/sglang/srt/model_executor/cuda_graph_runner.py @@ -116,7 +116,7 @@ def get_batch_sizes_to_capture(model_runner: ModelRunner): if capture_bs is None: if server_args.speculative_algorithm is None: if server_args.disable_cuda_graph_padding: - capture_bs = list(range(1, 33)) + range(40, 161, 16) + capture_bs = list(range(1, 33)) + list(range(40, 161, 16)) else: capture_bs = [1, 2, 4, 8] + list(range(16, 161, 8)) else: