server : do not release slot on image input (#3798)

This commit is contained in:
Georgi Gerganov 2023-10-26 22:53:37 +03:00
parent 6961c4bd0b
commit 34b2a5e1ee
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -1502,7 +1502,7 @@ struct llama_server_context
{ {
for (auto & slot : slots) for (auto & slot : slots)
{ {
const bool has_prompt = slot.prompt.is_array() || (slot.prompt.is_string() && !slot.prompt.get<std::string>().empty()); const bool has_prompt = slot.prompt.is_array() || (slot.prompt.is_string() && !slot.prompt.get<std::string>().empty()) || !slot.images.empty();
// empty prompt passed -> release the slot and send empty response // empty prompt passed -> release the slot and send empty response
if (slot.state == IDLE && slot.command == LOAD_PROMPT && !has_prompt) if (slot.state == IDLE && slot.command == LOAD_PROMPT && !has_prompt)