root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-11-13 14:29:52 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

llava-cli : don't crash if --image flag is invalid (#4835)

Browse Source 
This change fixes an issue where supplying `--image missing-file` would
result in a segfault due to a null pointer being dereferenced. This can
result in distracting info being printed if robust crash analysis tools
are being used.
This commit is contained in:
Justine Tunney 2024-01-09 09:59:14 -08:00 committed by GitHub
parent 4dccb38d9a
commit 36e5a08b20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/llava/llava-cli.cpp
View File

@ -243,6 +243,9 @@ int main(int argc, char ** argv) {
} }
auto image_embed = load_image(ctx_llava, &params); auto image_embed = load_image(ctx_llava, &params);
if (!image_embed) {
return 1;
}
// process the prompt // process the prompt
process_prompt(ctx_llava, image_embed, &params, params.prompt); process_prompt(ctx_llava, image_embed, &params, params.prompt);