batched: fix n_predict parameter (#8527)

This commit is contained in:
Masaya, Kato 2024-07-17 16:34:28 +09:00 committed by GitHub
parent d65a8361fe
commit da3913d8f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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