Skip to content

Commit

Permalink
main : switch input_noecho to input_echo to remove negation (ggml-org…
Browse files Browse the repository at this point in the history
…#979)

Signed-off-by: deadprogram <ron@hybridgroup.com>
  • Loading branch information
deadprogram authored May 2, 2023
1 parent 2d099e5 commit e2cd506
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions examples/main/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,7 @@ int main(int argc, char ** argv) {
}

bool is_antiprompt = false;
bool input_noecho = false;
bool input_echo = true;

// HACK - because session saving incurs a non-negligible delay, for now skip re-saving session
// if we loaded a session with at least 75% similarity. It's currently just used to speed up the
Expand Down Expand Up @@ -485,7 +485,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;
Expand All @@ -503,14 +503,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);
}

Expand Down Expand Up @@ -605,7 +605,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) {
Expand Down

0 comments on commit e2cd506

Please sign in to comment.