[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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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,