CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-29 08:41:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

llava : avoid segfault in case of non-existent mmproj file (#3674)

Browse Source
This commit is contained in:
M. Yusuf Sarıgöz
2023-10-19 16:59:11 +03:00
committed by GitHub
parent 004797f6ac
commit 60abea9798
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/llava/clip.cpp
View File

@@ -112,8 +112,7 @@ static float get_f32(const gguf_context * ctx, const std::string & key) {
static struct ggml_tensor * get_tensor(struct ggml_context * ctx, const std::string & name) { static struct ggml_tensor * get_tensor(struct ggml_context * ctx, const std::string & name) {
struct ggml_tensor * cur = ggml_get_tensor(ctx, name.c_str()); struct ggml_tensor * cur = ggml_get_tensor(ctx, name.c_str());
if (!cur) { if (!cur) {
printf("unable to find tensor %s\n", name.c_str()); throw std::runtime_error(format("%s: unable to find tensor %s\n", __func__, name.c_str()));
throw std::runtime_error(format("unable to find tensor %s\n", name.c_str()));
} }
return cur; return cur;
@@ -136,7 +135,7 @@ static std::string get_ftype(int ftype) {
case 8: case 8:
return "q8_0"; return "q8_0";
default: default:
throw std::runtime_error(format("Unrecognized file type: %d\n", ftype)); throw std::runtime_error(format("%s: Unrecognized file type: %d\n", __func__, ftype));
} }
} }
@@ -462,6 +461,9 @@ struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
}; };
struct gguf_context * ctx = gguf_init_from_file(fname, params); struct gguf_context * ctx = gguf_init_from_file(fname, params);
if (!ctx) {
throw std::runtime_error(format("%s: failed to load CLIP model from %s. Does this file exist?\n", __func__, fname));
}
if (verbosity >= 1) { if (verbosity >= 1) {
const int n_tensors = gguf_get_n_tensors(ctx); const int n_tensors = gguf_get_n_tensors(ctx);