Prechádzať zdrojové kódy

fix: check model pointer validity before use (#13631)

Nick 8 mesiacov pred
rodič
commit
9c55e5c5c2
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  1. 1 1
      examples/simple/simple.cpp

+ 1 - 1
examples/simple/simple.cpp

@@ -84,13 +84,13 @@ int main(int argc, char ** argv) {
     model_params.n_gpu_layers = ngl;
     model_params.n_gpu_layers = ngl;
 
 
     llama_model * model = llama_model_load_from_file(model_path.c_str(), model_params);
     llama_model * model = llama_model_load_from_file(model_path.c_str(), model_params);
-    const llama_vocab * vocab = llama_model_get_vocab(model);
 
 
     if (model == NULL) {
     if (model == NULL) {
         fprintf(stderr , "%s: error: unable to load model\n" , __func__);
         fprintf(stderr , "%s: error: unable to load model\n" , __func__);
         return 1;
         return 1;
     }
     }
 
 
+    const llama_vocab * vocab = llama_model_get_vocab(model);
     // tokenize the prompt
     // tokenize the prompt
 
 
     // find the number of tokens in the prompt
     // find the number of tokens in the prompt