fix incorrect if branch

This commit is contained in:
Xuan Son Nguyen 2024-10-23 23:48:08 +02:00
parent 3abc33962e
commit 60d4194bfe

View file

@ -2009,7 +2009,6 @@ struct server_context {
}
slot.n_prompt_tokens_processed = 0;
}
// non-causal tasks require to fit the entire prompt in the physical batch
if (slot.cmpl_type == SERVER_TASK_CMPL_TYPE_EMBEDDING || slot.cmpl_type == SERVER_TASK_CMPL_TYPE_RERANK) {
@ -2079,6 +2078,7 @@ struct server_context {
SLT_INF(slot, "prompt done, n_past = %d, n_tokens = %d\n", slot.n_past, batch.n_tokens);
}
}
if (batch.n_tokens >= n_batch) {
break;