We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6f80ea1 commit 850c6c2Copy full SHA for 850c6c2
csrc/gpu/update_inputs_v2.cu
@@ -60,10 +60,11 @@ __global__ void update_inputs_kernel_v2(
60
if (stop_flags[thread_idx]) {
61
if (seq_lens_this_time[thread_idx] == 0) {
62
next_tokens[thread_idx] = -1;
63
- } else {
64
- next_tokens[thread_idx] = end_ids[0];
65
- kwargs_next_tokens[thread_idx] = end_ids[0];
66
}
+ // else {
+ // next_tokens[thread_idx] = end_ids[0];
+ // kwargs_next_tokens[thread_idx] = end_ids[0];
67
+ // }
68
} else {
69
kwargs_next_tokens[thread_idx] = next_tokens[thread_idx];
70
0 commit comments