diff --git a/python/sglang/bench_one_batch.py b/python/sglang/bench_one_batch.py index 120eb04ee..acf1520ed 100644 --- a/python/sglang/bench_one_batch.py +++ b/python/sglang/bench_one_batch.py @@ -207,7 +207,7 @@ def prepare_extend_inputs_for_correctness_test( def prepare_synthetic_inputs_for_latency_test(batch_size, input_len): - input_ids = np.ones((batch_size, input_len), dtype=np.int32) + input_ids = np.random.randint(0, 10000, (batch_size, input_len), dtype=np.int32) sampling_params = SamplingParams( temperature=0, max_new_tokens=BenchArgs.output_len,