From 1c931f3d4f7b7b296d588e19e738ae65006791eb Mon Sep 17 00:00:00 2001
From: Pierrick Hymbert <pierrick.hymbert@gmail.com>
Date: Thu, 21 Mar 2024 20:50:28 +0100
Subject: [PATCH] Handle optional tensors

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
---
 llama.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/llama.cpp b/llama.cpp
index 21f6ad761..b1e2e062c 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -3114,7 +3114,11 @@ struct llama_model_loader {
     }
 
     struct ggml_tensor * get_tensor_meta(const char * name) const {
-        return get_weights(name).tensor;
+        try {
+            return get_weights(name).tensor;
+        } catch (const std::runtime_error & e) {
+            return NULL;
+        }
     }
 
     struct ggml_tensor * get_tensor_meta(int i) const {