server : fix EOS token detection with disabled cache (#5938)

This commit is contained in:
Georgi Gerganov 2024-03-08 12:40:02 +02:00 committed by GitHub
parent 581ed5c4fe
commit af37fd8b30
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1123,7 +1123,7 @@ struct server_context {
}); });
} }
if (!slot.cache_tokens.empty() && result.tok == llama_token_eos(model)) { if (result.tok == llama_token_eos(model)) {
slot.stopped_eos = true; slot.stopped_eos = true;
slot.has_next_token = false; slot.has_next_token = false;