This commit is contained in:
netrunnereve 2023-08-06 00:40:13 -04:00
parent 215e2f21d0
commit ce6d86ec41

View file

@ -126,7 +126,7 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
} }
params.pp_threads = std::stoi(argv[i]); params.pp_threads = std::stoi(argv[i]);
if (params.pp_threads <= 0) { if (params.pp_threads <= 0) {
params.pp_threads = params.n_threads; params.pp_threads = std::thread::hardware_concurrency();
} }
} else if (arg == "-p" || arg == "--prompt") { } else if (arg == "-p" || arg == "--prompt") {
if (++i >= argc) { if (++i >= argc) {