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

examples : switch input_noecho to input_echo to remove negation

Browse source 
Signed-off-by: deadprogram <ron@hybridgroup.com>
This commit is contained in:
deadprogram 2023-04-25 19:55:25 +02:00
parent dd0eabc049
commit 618fda5009
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/main/main.cpp
View file

@ -250,7 +250,7 @@ int main(int argc, char ** argv) {
} }
bool is_antiprompt = false; bool is_antiprompt = false;
bool input_noecho = false; bool input_echo = true;
int n_past = 0; int n_past = 0;
int n_remain = params.n_predict; int n_remain = params.n_predict;
@ -340,7 +340,7 @@ int main(int argc, char ** argv) {
embd.push_back(id); embd.push_back(id);
// echo this to console // echo this to console
input_noecho = false; input_echo = true;
// decrement remaining sampling budget // decrement remaining sampling budget
--n_remain; --n_remain;
@ -358,14 +358,14 @@ int main(int argc, char ** argv) {
} }
// display text // display text
if (!input_noecho) { if (input_echo) {
for (auto id : embd) { for (auto id : embd) {
printf("%s", llama_token_to_str(ctx, id)); printf("%s", llama_token_to_str(ctx, id));
} }
fflush(stdout); fflush(stdout);
} }
// reset color to default if we there is no pending user input // reset color to default if we there is no pending user input
if (!input_noecho && (int)embd_inp.size() == n_consumed) { if (input_echo && (int)embd_inp.size() == n_consumed) {
set_console_color(con_st, CONSOLE_COLOR_DEFAULT); set_console_color(con_st, CONSOLE_COLOR_DEFAULT);
} }
@ -460,7 +460,7 @@ int main(int argc, char ** argv) {
n_remain -= line_inp.size(); n_remain -= line_inp.size();
} }
input_noecho = true; // do not echo this again input_echo = false; // do not echo this again
} }
if (n_past > 0) { if (n_past > 0) {