Jelajahi Sumber

llm: add back check for empty token cache

jmorganca 1 tahun lalu
induk
melakukan
fcf4d60eee
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      llm/ext_server/server.cpp

+ 1 - 1
llm/ext_server/server.cpp

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