Skip to content

Commit 7a87d31

Browse files
authored
[main] fix infinite generation (-n == -1) (#523)
1 parent 348d692 commit 7a87d31

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/main/main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ int main(int argc, char ** argv) {
493493
}
494494

495495
// In interactive mode, respect the maximum number of tokens and drop back to user input when reached.
496-
if (params.interactive && n_remain <= 0) {
496+
if (params.interactive && n_remain <= 0 && params.n_predict != -1) {
497497
n_remain = params.n_predict;
498498
is_interacting = true;
499499
}

0 commit comments

Comments
 (0)