tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-07-27 03:33:46 -04:00
Code Issues Packages Projects Releases Wiki Activity

llama : fix --reverse-prompt crashing issue (#14794)

Browse Source 
Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
This commit is contained in:
Molly Sophia
2025-07-21 17:38:36 +08:00
committed by GitHub
parent b4efd77f8a
commit c82d48ec23
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tools/main/main.cpp
View File

@ -785,6 +785,8 @@ int main(int argc, char ** argv) {
} }
// check for reverse prompt using special tokens // check for reverse prompt using special tokens
// avoid calling common_sampler_last() if last_output is empty
if (!last_output.empty()) {
llama_token last_token = common_sampler_last(smpl); llama_token last_token = common_sampler_last(smpl);
for (auto token : antiprompt_token) { for (auto token : antiprompt_token) {
if (token == last_token) { if (token == last_token) {
@ -795,6 +797,7 @@ int main(int argc, char ** argv) {
break; break;
} }
} }
}
if (is_antiprompt) { if (is_antiprompt) {
LOG_DBG("found antiprompt: %s\n", last_output.c_str()); LOG_DBG("found antiprompt: %s\n", last_output.c_str());