send partial stop string when <EOG> is reached

matteoserva · matteoserva · commit 1c4809a6568c · 2025-08-01T10:22:07.000+02:00
diff --git a/tools/server/server.cpp b/tools/server/server.cpp
@@ -2319,7 +2319,7 @@ struct server_context {
                     slot.generated_text.begin() + pos + stop_pos,
                     slot.generated_text.end());
                 pos = std::min(slot.n_sent_text, slot.generated_text.size());
-            } else if (slot.has_next_token) {
+            } else if (slot.has_next_token && !llama_vocab_is_eog(vocab, result.tok) ) {
                 stop_pos = slot.find_stopping_strings(str_test, token_str.size(), false);
                 send_text = stop_pos == std::string::npos;
             }

Original file line number	Diff line number	Diff line change
`@@ -2319,7 +2319,7 @@ struct server_context {`
`2319`	`2319`	`slot.generated_text.begin() + pos + stop_pos,`
`2320`	`2320`	`slot.generated_text.end());`
`2321`	`2321`	`pos = std::min(slot.n_sent_text, slot.generated_text.size());`
`2322`		`- } else if (slot.has_next_token) {`
	`2322`	`+ } else if (slot.has_next_token && !llama_vocab_is_eog(vocab, result.tok) ) {`
`2323`	`2323`	`stop_pos = slot.find_stopping_strings(str_test, token_str.size(), false);`
`2324`	`2324`	`send_text = stop_pos == std::string::npos;`
`2325`	`2325`	`}`