From: marcoStocchi <redacted>
Date: Fri, 21 Mar 2025 09:12:45 +0000 (+0100)
Subject: llama-tts : avoid crashes related to bad model file paths (#12482)
X-Git-Tag: upstream/0.0.5028~94
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=ea1518e839abe668c20f7e0074c0721f803da898;p=pkg%2Fggml%2Fsources%2Fllama.cpp

llama-tts : avoid crashes related to bad model file paths (#12482)
---

diff --git a/examples/tts/tts.cpp b/examples/tts/tts.cpp
index d953cadd..4cc42e16 100644
--- a/examples/tts/tts.cpp
+++ b/examples/tts/tts.cpp
@@ -571,6 +571,10 @@ int main(int argc, char ** argv) {
     model_ttc = llama_init_ttc.model.get();
     ctx_ttc   = llama_init_ttc.context.get();
 
+    if (model_ttc == nullptr || ctx_ttc == nullptr) {
+        return ENOENT;
+    }
+
     const llama_vocab * vocab = llama_model_get_vocab(model_ttc);
 
     // TODO: refactor in a common struct
@@ -586,6 +590,10 @@ int main(int argc, char ** argv) {
     model_cts = llama_init_cts.model.get();
     ctx_cts   = llama_init_cts.context.get();
 
+    if (model_cts == nullptr || ctx_cts == nullptr) {
+        return ENOENT;
+    }
+
     std::vector<common_sampler *> smpl(n_parallel);
     for (int i = 0; i < n_parallel; ++i) {
         params.sampling.no_perf = (i != 0);