vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

minor : add braces

Browse source
This commit is contained in:
Georgi Gerganov 2023-12-06 10:04:49 +02:00 • committed by GitHub
parent 8f665eaddc
commit b2e8a2e22f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
examples/speculative/speculative.cpp
View file

@ -203,8 +203,9 @@ int main(int argc, char ** argv) {
const std::string token_str = llama_token_to_piece(ctx_tgt, id); const std::string token_str = llama_token_to_piece(ctx_tgt, id);
if (!params.use_color) if (!params.use_color) {
printf("%s", token_str.c_str()); printf("%s", token_str.c_str());
}
if (id == llama_token_eos(model_tgt)) { if (id == llama_token_eos(model_tgt)) {
has_eos = true; has_eos = true;
@ -244,8 +245,9 @@ int main(int argc, char ** argv) {
continue; continue;
} }
} }
if (params.use_color) if (params.use_color) {
printf("%s", token_str.c_str()); printf("%s", token_str.c_str());
}
fflush(stdout); fflush(stdout);
LOG("the sampled target token (%d, '%s') did not match, or we ran out of drafted tokens\n", id, token_str.c_str()); LOG("the sampled target token (%d, '%s') did not match, or we ran out of drafted tokens\n", id, token_str.c_str());