ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-07 19:18:33 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

fix(llama.cpp): enable cont batching when parallel is set (#1622)

Browse Source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-01-21 14:59:48 +01:00 committed by GitHub
parent 94261b1717
commit 697c769b64
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/cpp/llama/grpc-server.cpp
View File

@ -2465,10 +2465,10 @@ static void params_parse(const backend::ModelOptions* request,
const char *env_parallel = std::getenv("LLAMACPP_PARALLEL"); const char *env_parallel = std::getenv("LLAMACPP_PARALLEL");
if (env_parallel != NULL) { if (env_parallel != NULL) {
params.n_parallel = std::stoi(env_parallel); params.n_parallel = std::stoi(env_parallel);
params.cont_batching = true;
} else { } else {
params.n_parallel = 1; params.n_parallel = 1;
} }
params.cont_batching = true;
// TODO: Add yarn // TODO: Add yarn
if (!request->tensorsplit().empty()) { if (!request->tensorsplit().empty()) {