Skip to content

Commit 850c6c2

Browse files
authored
output_len truncated without eos_token_id (#10614)
1 parent 6f80ea1 commit 850c6c2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

csrc/gpu/update_inputs_v2.cu

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,11 @@ __global__ void update_inputs_kernel_v2(
6060
if (stop_flags[thread_idx]) {
6161
if (seq_lens_this_time[thread_idx] == 0) {
6262
next_tokens[thread_idx] = -1;
63-
} else {
64-
next_tokens[thread_idx] = end_ids[0];
65-
kwargs_next_tokens[thread_idx] = end_ids[0];
6663
}
64+
// else {
65+
// next_tokens[thread_idx] = end_ids[0];
66+
// kwargs_next_tokens[thread_idx] = end_ids[0];
67+
// }
6768
} else {
6869
kwargs_next_tokens[thread_idx] = next_tokens[thread_idx];
6970
}

0 commit comments

Comments
 (0)