diff --git a/tests/test-model-load-cancel.cpp b/tests/test-model-load-cancel.cpp index ff24a5955..cb3c012b9 100644 --- a/tests/test-model-load-cancel.cpp +++ b/tests/test-model-load-cancel.cpp @@ -1,8 +1,20 @@ #include "llama.h" +#include #include 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; }