fix the error of no ending

This commit is contained in:
Liu Ming 2023-05-29 08:37:34 +08:00
parent 991002b28e
commit 530eb57fe4

View file

@ -455,12 +455,14 @@ public:
{
has_next_token = false;
}
else
{
if (params.interactive && n_remain <= 0 && params.n_predict != -1)
{
n_remain = params.n_predict;
}
has_next_token = n_remain != 0;
}
return result;
}