[Benchmark][Bugfix] Fix Dataset Length Calculation (#19868)

Signed-off-by: Robert Shaw <robshaw@redhat.com>
Co-authored-by: Robert Shaw <robshaw@redhat.com>
This commit is contained in:
Robert Shaw
2025-06-19 21:30:41 -04:00
committed by GitHub
parent ea10dd9d9e
commit 10d82f9ac5

View File

@@ -353,7 +353,7 @@ class RandomDataset(BenchmarkDataset):
: input_lens[i]
]
prompt = tokenizer.decode(re_encoded_sequence)
total_input_len = prefix_len + int(input_lens[i])
total_input_len = len(re_encoded_sequence)
requests.append(
SampleRequest(
prompt=prompt,