diff --git a/benchmarks/benchmark_dataset.py b/benchmarks/benchmark_dataset.py index 8df071d6033fa..1ad6cef7a9dbc 100644 --- a/benchmarks/benchmark_dataset.py +++ b/benchmarks/benchmark_dataset.py @@ -324,6 +324,9 @@ class RandomDataset(BenchmarkDataset): input_low = int(real_input_len * (1 - range_ratio)) input_high = int(real_input_len * (1 + range_ratio)) output_low = int(output_len * (1 - range_ratio)) + # Ensure the lower bound for output length is at least 1 to prevent + # sampling 0 tokens, which can cause request failures. + output_low = max(output_low, 1) output_high = int(output_len * (1 + range_ratio)) # Add logging for debugging