From ea1518e839abe668c20f7e0074c0721f803da898 Mon Sep 17 00:00:00 2001 From: marcoStocchi Date: Fri, 21 Mar 2025 10:12:45 +0100 Subject: [PATCH] llama-tts : avoid crashes related to bad model file paths (#12482) --- examples/tts/tts.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/examples/tts/tts.cpp b/examples/tts/tts.cpp index d953cadd6..4cc42e167 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 smpl(n_parallel); for (int i = 0; i < n_parallel; ++i) { params.sampling.no_perf = (i != 0);