mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-20 03:25:37 +00:00
Set thread counts after loading model (#836)
This commit is contained in:
parent
fdffad9efe
commit
44dc1ade62
@ -24,11 +24,11 @@ void* load_model(const char *fname, int n_threads) {
|
|||||||
__func__, new_error.message);
|
__func__, new_error.message);
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
llmodel_setThreadCount(model, n_threads);
|
|
||||||
if (!llmodel_loadModel(model, fname)) {
|
if (!llmodel_loadModel(model, fname)) {
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
llmodel_setThreadCount(model, n_threads);
|
||||||
return model;
|
return model;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user