haraldwolff/llama.cpp
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Fail test if model file is missing

Browse source
This commit is contained in:
crasm 2023-12-17 14:31:03 -05:00
parent 4b1f70cb03
commit 32ebd525bf
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test-model-load-cancel.cpp
View file

@ -1,8 +1,20 @@
#include "llama.h"
#include <cstdio>
#include <cstdlib>
int main(void) {
auto model_path = "models/7B/ggml-model-f16.gguf";
auto file = fopen(model_path, "r");
if (file == nullptr) {
fprintf(stderr, "no model at '%s' found\n", model_path);
return EXIT_FAILURE;
} else {
fprintf(stderr, "using '%s'\n", model_path);
fclose(file);
}
llama_backend_init(false);
auto params = llama_model_params{};
params.use_mmap = false;
@ -10,7 +22,7 @@ int main(void) {
(void) ctx;
return progress > 0.50;
};
auto * model = llama_load_model_from_file("models/7B/ggml-model-f16.gguf", params);
auto * model = llama_load_model_from_file(model_path, params);
llama_backend_free();
return model == nullptr ? EXIT_SUCCESS : EXIT_FAILURE;
}