Kaynağa Gözat

llm: add back check for empty token cache

jmorganca 1 yıl önce
ebeveyn
işleme
fcf4d60eee
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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;