aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/main.cpp b/main.cpp
index 387d35f..d1defe2 100644
--- a/main.cpp
+++ b/main.cpp
@@ -976,6 +976,8 @@ int main(int argc, char ** argv) {
std::vector<gpt_vocab::id> line_inp = ::llama_tokenize(vocab, buf, false);
embd_inp.insert(embd_inp.end(), line_inp.begin(), line_inp.end());
+ remaining_tokens -= line_inp.size();
+
input_noecho = true; // do not echo this again
}