0003-load_exception.patch 1.82 KB
Newer Older
1
2
3
4
From 4f2b9cd0f012c49f40d0784454864ad41ca418b2 Mon Sep 17 00:00:00 2001
From: Gabe Goodhart <ghart@us.ibm.com>
Date: Thu, 19 Sep 2024 17:00:28 -0600
Subject: [PATCH] load exception
Michael Yang's avatar
Michael Yang committed
5
6
7
8
9

---
 src/llama.cpp | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

10
diff --git a/src/llama.cpp b/src/llama.cpp
11
index af8afd84..4d1db3d5 100644
12
13
--- a/src/llama.cpp
+++ b/src/llama.cpp
14
@@ -8871,7 +8871,7 @@ static int llama_model_load(const std::string & fname, llama_model & model, llam
15
16
17
18
19
20
21
         }
     } catch (const std::exception & err) {
         LLAMA_LOG_ERROR("%s: error loading model: %s\n", __func__, err.what());
-        return -1;
+        throw;
     }
 
22
23
     // loading time will be recalculate after the first eval, so
@@ -18675,16 +18675,23 @@ struct llama_model * llama_load_model_from_file(
Michael Yang's avatar
Michael Yang committed
24
25
         }
         model->rpc_servers.push_back(servers);
26
27
28
29
30
31
32
33
     }
-    int status = llama_model_load(path_model, *model, params);
-    GGML_ASSERT(status <= 0);
-    if (status < 0) {
-        if (status == -1) {
-            LLAMA_LOG_ERROR("%s: failed to load model\n", __func__);
-        } else if (status == -2) {
-            LLAMA_LOG_INFO("%s: cancelled model load\n", __func__);
Michael Yang's avatar
Michael Yang committed
34
+
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
+    try {
+        int status = llama_model_load(path_model, *model, params);
+        GGML_ASSERT(status <= 0);
+        if (status < 0) {
+            if (status == -1) {
+                LLAMA_LOG_ERROR("%s: failed to load model\n", __func__);
+            } else if (status == -2) {
+                LLAMA_LOG_INFO("%s: cancelled model load\n", __func__);
+            }
+            delete model;
+            return nullptr;
         }
+    } catch (...) {
+        LLAMA_LOG_ERROR("%s: exception loading model\n", __func__);
         delete model;
-        return nullptr;
+        throw;
     }
 
     return model;
Michael Yang's avatar
Michael Yang committed
55
-- 
56
2.39.3 (Apple Git-146)
Michael Yang's avatar
Michael Yang committed
57