Skip to content

Commit

Permalink
Comment
Browse files Browse the repository at this point in the history
  • Loading branch information
mgoin committed Dec 2, 2024
1 parent f066b25 commit c2b3b3a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion benchmarks/benchmark_throughput.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,8 @@ def main(args: argparse.Namespace):
diff = args.input_len - tokenized_len
if diff > 0:
candidate_ids.extend([
random.randint(0, vocab_size - 1) for _ in range(diff)
random.randint(100, vocab_size - 100)
for _ in range(diff)
])
else:
candidate_ids = candidate_ids[:diff]
Expand Down

0 comments on commit c2b3b3a

Please sign in to comment.