Attempted fix for main example getting stuck when -n -2 and --interactive

This commit is contained in:
KerfuffleV2 2023-08-24 09:09:49 -06:00
parent ef955fbd23
commit e979cef2ff

View file

@ -798,7 +798,7 @@ int main(int argc, char ** argv) {
} }
// In interactive mode, respect the maximum number of tokens and drop back to user input when reached. // In interactive mode, respect the maximum number of tokens and drop back to user input when reached.
if (params.interactive && n_remain <= 0 && params.n_predict != -1) { if (params.interactive && n_remain <= 0 && params.n_predict >= 0) {
n_remain = params.n_predict; n_remain = params.n_predict;
is_interacting = true; is_interacting = true;
} }