Don't crash when --image argument is invalid

Mozilla-Ocho · Jan 9, 2024 · c617679 · c617679
1 parent 64d1e65
commit c617679
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llama.cpp/llava/llava-cli.cpp b/llama.cpp/llava/llava-cli.cpp
@@ -263,6 +263,9 @@ int llava_cli(int argc, char ** argv) {
     sigaction(SIGINT, &sa, NULL);
 
     auto image_embed = load_image(ctx_llava, &params);
+    if (!image_embed) {
+        exit(1);
+    }
 
     // process the prompt
     process_prompt(ctx_llava, image_embed, &params, params.prompt);