batched: fix n_predict parameter (#8527)

This commit is contained in:
Masaya, Kato 2024-07-17 16:34:28 +09:00 committed by arthw
parent 050ba94e38
commit 8dad318d6f

View file

@ -31,7 +31,7 @@ int main(int argc, char ** argv) {
int n_parallel = params.n_parallel;
// total length of the sequences including the prompt
int n_predict = 32;
int n_predict = params.n_predict;
// init LLM