CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-11-03 09:22:01 +00:00
Code Issues 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
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;
}