batched: fix n_predict parameter

This commit is contained in:
msy-kato 2024-07-17 12:38:47 +09:00
parent 5e116e8dd5
commit 5b393836a9

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