Reset token budget after every user intervention.
In interactive mode, every time the model has to respond to user input it has an increasingly reduced token budget, eventually generating only a few words before stopping. The token budget in interactive should apply to every batch of tokens after user intervention, not globally
This commit is contained in:
parent
da5303c1ea
commit
dd459b604f
1 changed files with 2 additions and 2 deletions
4
main.cpp
4
main.cpp
|
@ -1054,12 +1054,12 @@ int main(int argc, char ** argv) {
|
||||||
embd_inp.insert(embd_inp.end(), inp_sfx.begin(), inp_sfx.end());
|
embd_inp.insert(embd_inp.end(), inp_sfx.begin(), inp_sfx.end());
|
||||||
}
|
}
|
||||||
|
|
||||||
remaining_tokens -= line_inp.size();
|
remaining_tokens = params.n_predict - line_inp.size();
|
||||||
|
|
||||||
input_noecho = true; // do not echo this again
|
input_noecho = true; // do not echo this again
|
||||||
}
|
|
||||||
is_interacting = false;
|
is_interacting = false;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// end of text token
|
// end of text token
|
||||||
if (embd.back() == EOS_TOKEN_ID) {
|
if (embd.back() == EOS_TOKEN_ID) {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue