sleepy/llama.cpp
1
0
Fork 0
Code Issues 11 Pull Requests Actions 108 Packages Projects Releases Wiki Activity

common: add bounds check in common_init_result::sampler to prevent segfault on failed model load (#21082)

Browse Source 
* common: add bounds check in common_init_result::sampler to prevent segfault on failed model load

* Revert a308e584ca

* Add regression test

* Remove regression test for init-fail sampler check
This commit is contained in:
mtmcp
2026-03-31 07:04:42 -03:00
committed by GitHub
parent fcc2d598c8
commit 90aa83c6bd
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/completion/completion.cpp
+1 -7
View File
@@ -146,19 +146,13 @@ int main(int argc, char ** argv) {
ctx = llama_init->context();
model = llama_init->model();
smpl = llama_init->sampler(0);
if (ctx == NULL) {
LOG_ERR("%s: error: unable to create context\n", __func__);
return 1;
}
if (model == NULL) {
LOG_ERR("%s: error: unable to load model\n", __func__);
return 1;
}
smpl = llama_init->sampler(0);
llama_memory_t mem = llama_get_memory(ctx);
const llama_vocab * vocab = llama_model_get_vocab(model);