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 input_noecho = false;
bool input_echo = true;
int n_past = 0;
int n_remain = params.n_predict;
@ -340,7 +340,7 @@ int main(int argc, char ** argv) {
embd.push_back(id);
// echo this to console
input_noecho = false;
input_echo = true;
// decrement remaining sampling budget
--n_remain;
@ -358,14 +358,14 @@ int main(int argc, char ** argv) {
}
// display text
if (!input_noecho) {
if (input_echo) {
for (auto id : embd) {
printf("%s", llama_token_to_str(ctx, id));
}
fflush(stdout);
}
// 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);
}
@ -460,7 +460,7 @@ int main(int argc, char ** argv) {
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) {