Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "0920ab9131274df143cfc49245409378a009b3c6"
Unverified Commit 10d82f9a authored by Robert Shaw's avatar Robert Shaw Committed by GitHub
Browse files

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


Signed-off-by: default avatarRobert Shaw <robshaw@redhat.com>
Co-authored-by: default avatarRobert Shaw <robshaw@redhat.com>
parent ea10dd9d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
benchmarks/benchmark_dataset.py
View file @ 10d82f9a
...@@ -353,7 +353,7 @@ class RandomDataset(BenchmarkDataset): ...@@ -353,7 +353,7 @@ class RandomDataset(BenchmarkDataset):
: input_lens[i] : input_lens[i]
] ]
prompt = tokenizer.decode(re_encoded_sequence) prompt = tokenizer.decode(re_encoded_sequence)
total_input_len = prefix_len + int(input_lens[i]) total_input_len = len(re_encoded_sequence)
requests.append( requests.append(
SampleRequest( SampleRequest(
prompt=prompt, prompt=prompt,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment