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

No --in-prefix coloring

Browse source 
The `--in-prefix` text was inconsistently colored. Now, it's never colored, just like the `--in-suffix` text.
This commit is contained in:
h-h-h-h 2023-10-01 11:58:36 +02:00
parent fd6b6b2426
commit 178d0dd78b
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
examples/main/main.cpp
View file

@ -694,10 +694,8 @@ int main(int argc, char ** argv) {
if (last_output.find(antiprompt, search_start_pos) != std::string::npos) { if (last_output.find(antiprompt, search_start_pos) != std::string::npos) {
if (params.interactive) { if (params.interactive) {
is_interacting = true; is_interacting = true;
console::set_display(console::user_input);
} }
is_antiprompt = true; is_antiprompt = true;
fflush(stdout);
break; break;
} }
} }
@ -721,8 +719,6 @@ int main(int argc, char ** argv) {
is_interacting = true; is_interacting = true;
printf("\n"); printf("\n");
console::set_display(console::user_input);
fflush(stdout);
} else if (params.instruct) { } else if (params.instruct) {
is_interacting = true; is_interacting = true;
} }
@ -747,6 +743,9 @@ int main(int argc, char ** argv) {
printf("%s", buffer.c_str()); printf("%s", buffer.c_str());
} }
// color user input only
console::set_display(console::user_input);
std::string line; std::string line;
bool another_line = true; bool another_line = true;
do { do {