Respect the maximum number of tokens in interactive. (#298)

Co-authored-by: Johnman <johnman@github>
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
pull/299/head master-368d0c8
tjohnman 1 year ago committed by GitHub
parent 50fae10d03
commit 368d0c8a9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1062,7 +1062,6 @@ int main(int argc, char ** argv) {
}
// end of text token
if (embd.back() == EOS_TOKEN_ID) {
if (params.interactive) {
is_interacting = true;
@ -1071,6 +1070,12 @@ int main(int argc, char ** argv) {
break;
}
}
// In interactive mode, respect the maximum number of tokens and drop back to user input when reached.
if (params.interactive && remaining_tokens <= 0) {
remaining_tokens = params.n_predict;
is_interacting = true;
}
}
#if defined (_WIN32)

Loading…
Cancel
Save